Deep Learning Compiler Verification Engineer

NVIDIA Corporation; Santa Clara (CA); On-site; USD 116,000 - 190,000; Full time

Job summary

NVIDIA Corporation is seeking an engineer to help build and verify compiler technology for AI workloads in Santa Clara, CA. You will develop verification software and infrastructure to ensure correctness across graph transformations, IR lowering, and GPU execution.

You will work with DL frameworks and compiler teams to validate optimizations, extend test-generation systems, and improve measurement of functional quality and performance as models and hardware evolve.

Qualifications

• BS, MS or PhD in Computer Science, Computer Engineering, Mathematics, or equivalent experience.
• 2+ years of hands-on software engineering experience, with experience in compiler development, deep learning systems, or compiler verification.
• Proficiency in Python or C++ and experience with one major DL framework (e.g., PyTorch, JAX/XLA, TensorRT) and model execution, graph representation, or runtime behavior.
• Strong problem-solving and debugging skills — ability to reason about model behavior and compiler-generated code, investigate correctness issues, and collaborate with teammates to resolve edge-case failures.

Responsibilities

• Develop and maintain components of systems to reason about correctness in deep learning compilers, across graph transformations, IR lowering, and GPU execution.
• Analyze and validate optimizations (graph rewrites in MLIR, fusion passes, mixed-precision transformations) to preserve semantics and numerical behavior.
• Implement and extend test generation systems that explore model topologies, precision modes, and hardware targets.
• Contribute to improving how we measure functional quality and performance as models, compiler stacks, and hardware evolve.

Skills

Compiler development

Deep learning systems

Compiler verification

Python

C++

MLIR

LLVM

Education

BS/MS/PhD in CS/CE/Math or equivalent

Tools

MLIR

LLVM

TVM

XLA

Job description

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work.

We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to implement compiler verification software & related infrastructure in the AI space. You will be solving critical problems working alongside a diverse set of minds in GPU computing and systems software, doing what you enjoy. If this sounds like a fun challenge, we want to hear from you.

What you'll be doing:

• Develop and maintain components of systems to reason about correctness in deep learning compilers, across graph transformations, IR lowering, and GPU execution.
• Work with deep learning compiler and architecture teams to analyze and validate optimizations (e.g., graph rewrites in MLIR, fusion passes, mixed-precision transformations), ensuring they preserve semantics and numerical behavior.
• Implement and extend test generation systems that use deep learning solutions and analysis methods to drive in-depth testing. These systems explore the vast combinatorial space of model topologies, precision modes, and hardware targets.
• Contribute to improving how we measure functional quality and performance as models, compiler stacks, and hardware continue to evolve.

What we need to see:

• BS, MS or PhD in Computer Science, Computer Engineering, Mathematics, or equivalent experience.
• 2+ years of hands-on software engineering experience, with experience in compiler development, deep learning systems, or compiler verification.
• Proficiency in Python or C++ and experience with one major DL framework. This could be PyTorch, JAX/XLA, TensorRT, or a similar framework. Experience should involve model execution, graph representation, or runtime behavior.
• Strong problem-solving and debugging skills — ability to reason about model behavior and compiler-generated code, investigate correctness issues, and collaborate with teammates to resolve edge-case failures.

Ways to stand out from the crowd:

• Experience with compiler technologies such as LLVM, MLIR, TVM, or XLA, and familiarity with compiler passes, IR semantics, or correctness testing.
• Exposure to formal methods or language specification: type systems, program semantics, or proof-based verification.
• Experience with DL model internals such as quantization, operator fusion, mixed-precision, or graph-level optimization.

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary range for this position is 116,000 USD – 189,750 USD.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 4, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.