Senior Compiler Verification Engineer - Deep Learning

Job Description

Engineer role in compiler verification for deep learning workloads in the AI space at NVIDIA.

Requirements

• 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 like JAX, PyTorch or TensorFlow
• Strong programming skills in Python
• Focused, quick learner with strong analytical skills and attention to detail
• Strong troubleshooting and debugging skills
• Creative problem-solving skills
• Experience with Large Language Models is a plus
• Experience testing production software or developing fuzzing tools for software testing is a plus
• Knowledge of other programming languages and domains such as CUDA is a plus
• Compiler knowledge is a plus

Responsibilities

• Work closely with compiler developers to verify new deep learning 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 weaknesses in the current process and suggest improvements for code coverage
• Participate in quality initiatives and drive continuous improvement

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching