AI DL Compiler Verification Engineer

NVIDIA Corporation; Seattle (WA); Vor Ort; USD 116.000 - 190.000; Vollzeit

Benefits dieser Stelle

Equity

Benefits

Zusammenfassung

NVIDIA Corporation in Seattle is seeking an engineer to implement compiler verification software and related infrastructure for AI workloads. You will solve critical problems alongside a diverse team in GPU computing and systems software, contributing to rigorously tested compiler pipelines.

The role requires a PhD or Master’s with 2+ years of software/ compiler work, strong Python or C++ skills, and experience with PyTorch or similar DL frameworks. Equity and benefits are part of the package.

Qualifikationen

• 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.
• 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.

Aufgaben

• 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.

Kenntnisse

Python or C++

Strong debugging

Compiler development

Ausbildung

BS/MS/PhD in Computer Science/Engineering/Mathematics

Tools

PyTorch

JAX/XLA

TensorRT

MLIR/LLVM

Jobbeschreibung

NVIDIA Corporation in Seattle is seeking an engineer to implement compiler verification software and related infrastructure for AI workloads. You will solve critical problems alongside a diverse team in GPU computing and systems software, contributing to rigorously tested compiler pipelines.

The role requires a PhD or Master’s with 2+ years of software/ compiler work, strong Python or C++ skills, and experience with PyTorch or similar DL frameworks. Equity and benefits are part of the package.