DL Compiler Verification Engineer for AI & GPU Systems

NVIDIA Corporation; Austin (TX); On-site; USD 116,000 - 190,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation in Austin, TX is seeking an engineer to implement compiler verification software in the AI space. You will solve critical problems alongside a diverse team in GPU computing and systems software, doing what you enjoy.

As part of the role, you will develop components to reason about correctness in deep learning compilers, analyze optimizations, and extend test generation systems that explore model topologies, precision modes, and hardware targets.

Qualifications

• 2+ years of hands-on software engineering, with compiler development or DL systems.
• Proficiency in Python or C++ and experience with a major DL framework.
• Strong problem-solving, debugging skills, and collaboration.

Responsibilities

• Develop and maintain components to reason about correctness in deep learning compilers.
• Analyze and validate optimizations ensuring semantics and numerical behavior.
• Extend test generation systems using DL approaches to explore model topologies and targets.
• Improve measurement of functional quality and performance as systems evolve.

Skills

Python

C++

Deep learning

Compiler development

Education

BS/MS/PhD in CS/CE/Math

Tools

PyTorch

JAX/XLA

TensorRT

Job description

NVIDIA Corporation in Austin, TX is seeking an engineer to implement compiler verification software in the AI space. You will solve critical problems alongside a diverse team in GPU computing and systems software, doing what you enjoy.

As part of the role, you will develop components to reason about correctness in deep learning compilers, analyze optimizations, and extend test generation systems that explore model topologies, precision modes, and hardware targets.