NVIDIA's invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as “the AI computing company”
Do you want to help drive the progress of compilers for machine learning applications? Are you excited to learn how GPU performance powers technology such as mobile gaming, deep learning, and self-driving cars? Are you passionate about challenging yourself and would you love to contribute as part of a world-class company? We are building the next generation of compiler technologies to accelerate deep learning workloads. We are looking for an engineer to work on compiler testing in the AI space. You will be solving critical problems working alongside some of the top valued diverse 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 test new and state of the art deep learning related features and components including crafting and executing unit, functional, and performance tests. This would include authoring and reviewing test 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 like JAX, PyTorch or TensorFlow
Strong programming skills in Python
You should be focused, learn quickly, and have strong analytical skills with attention to detail. Strong troubleshooting and debugging skills.
Demonstrated uses of creative thinking for solutions to exciting problems that matter.
Ways to stand out from the crowd:
Experience with Large Language Models
Experience testing production software, or developing fuzzing tools for software testing
Knowledge of other programming languages and domains such as CUDA.
Compiler knowledge: Have an understanding of how compilers work and how compilers 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!
The base salary range is 128,000 USD - 247,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.