Member of Mathematical Staff
About 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 guarantees the result. It's why we hold #1 on NVIDIA's own kernel benchmark across hundreds of production kernels. Our core proof machinery is real and it works, but right now there are some rough edges that limit the number of kernels we can prove. With the rise of AI capabilities in mathematics combined with our bit-identical models and tools, such as our own model for large-scale Lean proof writing, things that were nearly impossible just a few months ago are now totally viable, but we need someone who figures out what theory we're missing, builds it and then coordinates with the engineering team to deploy it at scale, which is why we are hiring a Member of Mathematical Staff to own the math behind the proof engine and other surfaces for applying rigorous math to an increasingly wishy-washy domain. This is not an advisory role and you'll have access to better tooling than anywhere else, such as bit-exact software models of hardware components like the tcgen05, a compiler capable of emitting code at a lower level than publicly possible, and frontier models with capabilities tailored to what your work needs, trained with our team.