Senior Compiler Engineer - AI
USonsitesenior$184K – $288K
Posted today · via Workday
About this role
NVIDIA's invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as "the AI computing company". NVIDIA is hiring world class Software Engineers with AI Compiler and distributed systems experience for its Deep Learning Compiler Technology team. The team builds JaxPP, NVIDIA's open-source library for pipeline parallelism in JAX. JaxPP turns a JAX training program into per-stage tasks and runs them across many GPUs.…
What we'd score you on
reqspace match rubricFive dimensions, recruiter-grade. Upload your resume and we'll generate a written explanation of where you fit and where the gaps are.
1
Skills match
For this role: python, c++, pytorch, jax, teams
2
Level fit
This role is senior-level. We check your trajectory against it.
3
Domain experience
Your work in the role's domain matters more than your years total. We weight recent and direct experience.
4
Recency
A skill you used last quarter weighs more than one from five years ago. We grade on recency, not lifetime.
5
Location fit
This role is based in US. We weight your proximity and willingness to relocate.
Score yourself on this role.
Free · no card · written explanation included
Skills in this role
Pulled from the job description. These are the keywords we'll weight when scoring your fit.
pythonc++pytorchjaxteams
More at NVIDIA
- View →LLM Reinforcement Learning Framework Intern - 2027China, Shanghai
- View →AI Compiler and Kernel Intern – Summer 2027US, TX, Austin
- View →Technical Product Marketing MBA Intern, AI Infrastructure - Summer 2027US, CA, Santa Clara
- View →Senior Manager, Material PlanningUS, CA, Santa Clara
- View →Senior AI EngineerUS, CA, Santa Clara
- View →Principal Engineer - AV Simulation and EvaluationUS, CA, Santa Clara