We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to work on compiler verification in the AI space. You will be solving critical problems working alongside a diverse set of minds in GPU computing and systems software, doing what you enjoy. If this sounds like a fun challenge, we want to hear from you!
What you’ll be doing:
In this role you will work closely with compiler developers to verify new and state of the art deep learning related features and components including crafting and executing unit, functional, and performance testing and benchmarking solutions. This would include authoring and reviewing verification plans, implementing & automating test cases and porting 3rd party tests. You will be generating test reports, isolating and classifying failures and tracking performance trends. You will help identify potential or observed weaknesses in the current process, offer ideas for actions that can improve code coverage, and participate in quality initiatives and drive continuous improvement.
What we need to see:
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, TensorRT or similar
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.
Ways to stand out from the crowd:
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.
Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. We have some of the most forward-thinking and hardworking people in the world working for us!
