Member of Technical Staff, Post-Training & Applied Research

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.

Speed at the kernel layer is only worth what we do with it though and our enterprise offering promises we'll turn a customer's dataset into a specialist model in days. To do that we run SFT, RL, DPO, design evals and distillation down to smaller and edge-deployable models. We're hiring a Member of Technical Staff to own the modeling side of that end to end.

The infrastructure you'll be working on top of is unusually strong, which is the only reason we're able to train models so fast. Every experiment goes through Model Foundry, which let's you run experiments on the best hardware, look at overviews or dive deep into any detail all in a versioned and reproducible manner, making it trivial to reuse and modify recipes. Underneath that is the most powerful engine you could ask for, which has been used to post-train multi-100B language parameter models on TPU, post-train robotics models on Trainium and pre-train AlphaFold v3 on AMD.