NVIDIA is looking for a Senior Engineer - Deep Learning Compiler Verification and Infrastructure to build robust verification software and infrastructure for deep learning workloads. At NVIDIA, you will collaborate directly with compiler developers to ensure the quality and performance of state-of-the-art components.

What You'll Do

Collaborate closely with compiler developers to verify new and advanced deep learning features and components.
Implement and execute functional and performance testing and benchmarking software.
Author and review verification plans, and build verification programs, scripts, and libraries.
Apply deep learning and other sophisticated techniques to create compiler verification solutions.
Identify weaknesses in the current process and propose actions to improve code coverage.
Participate in quality initiatives and drive continuous improvement.

What We're Looking For

BS or MS in Computer Science, Computer/Electrical Engineering, Mathematics, or related field (or equivalent experience).
3+ years programming experience in the Machine Learning domain, preferably using Python.
Experience with Deep Learning frameworks such as PyTorch, Scikit Learn, JAX/XLA, or TensorRT.
Ability to focus, learn quickly, and apply strong analytical skills with attention to detail.
Strong troubleshooting and debugging skills.
Proven use of creative thinking to develop solutions to complex problems.

Nice to Have

Experience with Large Language Models and applying deep learning to software engineering problems.
Hands-on compiler development or verification experience.
Knowledge of related technologies such as CUDA, Docker, and GPU-Accelerated Cloud.

Technical Stack

Python, PyTorch, Scikit Learn, JAX/XLA, TensorRT, CUDA, Docker

Benefits & Compensation

Compensation: $136,000 USD - $212,750 USD for Level 3, and $168,000 USD - $264,500 USD for Level 4.
Equity eligibility.
Comprehensive benefits package.

NVIDIA is committed to fostering a diverse work environment and is proud to be an equal opportunity employer. We do not discriminate 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.