NVIDIA is hiring a Senior SDET - Deep Learning Compilers to tackle compiler verification in AI, solving critical problems alongside experts in GPU computing and systems software. You will verify new and state-of-the-art deep learning compiler features and components.
What You'll Do
- Work closely with compiler developers to verify new and state-of-the-art deep learning related features and components.
- Craft and execute unit, functional, and performance testing and benchmarking solutions.
- Author and review verification plans, implement & automate test cases, and port 3rd party tests.
- Generate test reports, isolate and classify failures, and track performance trends.
- Identify potential or observed weaknesses in the current process and offer ideas for 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 Python programming experience in Machine Learning domain
- Experience working with Deep Learning frameworks such as JAX, PyTorch or TensorRT
- Focused, learn quickly, and have strong analytical skills with attention to detail
- Strong troubleshooting and debugging skills
- Proven uses of creative thinking for solutions to exciting problems that matter
Nice to Have
- Experience with Large Language Models
- Experience verifying production software, including developing software tools & solutions for software testing
- Knowledge of related programming languages and domains such as CUDA
- Compiler knowledge: Have an understanding of how compilers work and are implemented
Technical Stack
- Python, JAX, PyTorch, TensorRT, CUDA
Benefits & Compensation
- Highly competitive salaries
- Comprehensive benefits package
- Eligible for equity and benefits
- Compensation: 136,000 USD - 264,500 USD + equity
NVIDIA is committed to fostering a diverse 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.
