Senior SDeT- Deep Learning Compiler

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 verification 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 deep learning compiler developers to create software infrastructure and applications used to develop, test, and release compilers. Responsibilities involve creating and maintaining advanced CI/CD pipelines to deploy and monitor machine learning software workloads. This includes functional and performance benchmarking solutions used to verify and bring the quality of deep learning compilers at NVIDIA to the next level. In addition, you will evaluate the performance of AI models and systems by implementing and utilizing advanced algorithms and solutions to carry out meticulous testing, experimentation, and benchmarking. 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, MS, or PhD or equivalent experience in Computer Science, Computer/Electrical Engineering, Mathematics or related field

• 5+ years of overall professional experience with 2+ years of MLOps experience specific in deep learning software environment

• Knowledge of the machine learning domain and professional experience working with deep learning frameworks like TensorRT, JAX, PyTorch, NeMo, TensorFlow, or others.

• Able to architect, design, implement and debug complex solutions using Python programming language in machine learning domain

• Familiarity with CI/CD pipeline development, cloud-based testing (AWS/GCP), data storage and database solutions, implementing visualization dashboards, and data analysis for performance and regression reporting

• Learn and adapt swiftly to changes in technical and project requirements

Ways to stand out from the crowd:

• Experience identifying and solving challenging technical problems by programmatic use of deep learning solutions such as large language models

• Prior experience providing technical leadership in deep learning software development environments

• Knowledge on how compilers work and how compilers are implemented with LLVM infrastructure

With highly competitive salaries and a comprehensive benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most brilliant and hardworking people in the world working with us and our product lines are growing fast in some of the hottest state of the art fields such as Deep Learning, Artificial Intelligence, Autonomous Vehicles, Virtual Reality, etc. Our diverse team of talented, capable, and professional people are our greatest asset! If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you!

The base salary range is 136,000 USD - 264,500 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.

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.