Member of Technical Staff, AI-Driven Compilation

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 AI-Driven Compilation to own the search itself. Proving correctness at the end is what makes the search legal and your job is to make it good.

You'll be building the agentic and RL systems that decide which programs are worth trying (instruction selection, scheduling, barriers and stall counts, tiling, fusion, phase ordering) over a space that is enormous precisely because nothing in it has to be conservative.

The reward signal here is unusually clean for RL: