Innovative DL 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 implement compiler verification software and related infrastructure in the AI space. You will solve critical problems alongside a diverse team in GPU computing and systems software, focusing on correctness across graph transformations and runtime behavior.

The role involves developing test generation systems, validating DL compiler optimizations, and contributing to functional quality 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 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).
• 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, and 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.
• Work with deep learning compiler and architecture teams to analyze and validate optimizations, 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.
• 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 evolve.

Skills

Python

C++

Deep Learning

Compiler development

PyTorch

XLA

TensorRT

Graph representation

Runtime behavior

Education

Bachelor's/Master's/PhD in CS or related

Tools

LLVM

MLIR

TVM

XLA

Job description

NVIDIA Corporation is seeking an engineer to implement compiler verification software and related infrastructure in the AI space. You will solve critical problems alongside a diverse team in GPU computing and systems software, focusing on correctness across graph transformations and runtime behavior.

The role involves developing test generation systems, validating DL compiler optimizations, and contributing to functional quality as models and hardware evolve.