Engineering Manager, AI Compiler Analysis
$168k - $322k • Remote • US, CA, Santa Clara
Posted 3d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
Fully remote
Categories
LLM Engineer
About the job
NVIDIA is seeking an Engineering Manager to lead the strategy for verifying AI compilers for next-generation deep learning workloads. This role involves managing a technical team focused on AI compiler verification, ensuring the quality of AI compilers that optimize deep learning models for GPU execution. The position requires a deep understanding of compiler quality challenges at the intersection of compilers, machine learning frameworks, numerical computing, formal reasoning, and large-scale systems engineering.
Responsibilities
- Lead, mentor, and grow a team responsible for AI compiler verification.
- Own the verification of next-generation AI workloads, including LLMs and agentic AI systems, across the AI compiler stack and execution pipeline.
- Define formal-verification requirements for AI compiler transformations and generated GPU programs.
- Drive the use of AI-assisted and compiler-aware verification techniques.
- Partner with AI compiler development, CUDA software, ML framework, runtime, product, and AI software teams to build scalable verification infrastructure and improve engineering velocity.
Requirements
- BS, MS, or PhD in Computer Science, Computer Engineering, or related field, or equivalent experience.
- 10+ years of relevant software engineering experience.
- At least 3 years of experience leading engineering teams or major technical initiatives.
- Experience with AI compiler or framework technologies such as MLIR, TensorRT, XLA, Triton, PyTorch, or JAX.
- Fluency with AI workload and ML framework concepts, including computation graphs, tensor operations, model execution, and training or inference workflows.
- Strong people management skills, including hiring, mentoring, performance management, and team development.
Benefits
- Highly competitive salaries
- Comprehensive benefits package
- Equity
- Eligibility for Level 2 salary range: 168,000 USD - 270,250 USD
- Eligibility for Level 3 salary range: 200,000 USD - 322,000 USD