Member of Technical Staff at Wafer

Wafer's mission is to maximize intelligence per watt, by building AI that optimizes AI itself. Our journey starts with GPU kernels, but will expand into every corner of ML systems and AI infrastructure. We're a small team (4 people) backed by Fifty Years, Y Combinator, Jeff Dean, and Woj Zaremba (co-founder of OpenAI), and we're looking for engineers who want to work at the intersection of AI agents and systems programming. You'll work directly with the founding team to build the systems that power our GPU optimization platform, from the agent framework that iterates on kernels, to the profiling infrastructure that connects to NCU and ROCprofiler, to the compiler tooling that analyzes PTX and SASS. Build and improve our framework for GPU kernel optimization (multi-turn tool use, state management, reward signals) Build and improve our framework for GPU kernel optimization (multi-turn tool use, state management, reward signals) Develop integrations with GPU profilers and compiler toolchains Develop integrations with GPU profilers and compiler toolchains Design the architecture for remote GPU execution across cloud GPUs Design the architecture for remote GPU execution across cloud GPU