Member of Technical Staff, GPU Compiler
San Francisco Tensor Company, Bay Area
The role
We build the fastest GPU compiler in the world. Most compilers have to preserve correctness at every transform, constraining how far they can search, while we prove correctness at the end instead, allowing us to search a far wider space, with agents, with RL, with anything that works and still guarantee the result. It's why we hold #1 on NVIDIA's own kernel benchmark across hundreds of production kernels.
We're hiring a Member of Technical Staff for GPU Compiler Engineering to build the machine that does the searching.
You'll be working on the entire pipeline end to end (StableHLO ingestion, MLIR dialects & passes, backend code generation, executable GPU & host binaries). The pipeline already does things worth seeing. Given only a naive attention spec, the search found FlashAttention v4-level performance on a B300 in about 25 minutes and your job will be to make sure that next time, we beat FlashAttention with ease.