Senior Performance Engineer - AI for Science

For decades, NVIDIA has been at the forefront of accelerated computing, driving innovation in fields ranging from computer graphics to artificial intelligence. Today, we're leveraging this legacy to revolutionize digital biology and to accelerate drug discovery. Our cutting-edge GPU technology serves as the computational engine for groundbreaking research, enabling scientists to model complex biological systems, analyze and harmonize massive datasets, and accelerate the drug discovery process.

We're looking for a Senior Performance Engineer who shares our vision for transforming healthcare. This is a unique opportunity to apply your expertise in a fast-paced, collaborative environment where you'll be instrumental in developing the tools that will redefine what is possible in drug discovery. As part of our team, you'll work alongside researchers, developers, and experts in the field to tackle some of the most challenging and impactful problems of our time.

What you'll be doing:

• Optimizing the computational performance of the latest innovations in AI that address specific, domain-relevant bottlenecks.

• Designing and implementing computationally performant features for large scale, CUDA-backed ML training and inference frameworks, and data science tooling.

• Developing and maintaining an HPC software stack for generative machine learning models in digital biology and beyond

• Collaborating with multiple HPC, AI infrastructure, and research teams

• Driving the testing and maintenance of the algorithms and software modules

What we need to see:

• Advanced degree in a quantitative field such as Computer Science, Computational Biology, Computational Chemistry, Physics, Mathematics, or equivalent experience

• 8+ years of relevant experience

• Proven track record in performance engineering as well as software design, building and packaging and launching software products

• Deep understanding of parallel programming in C++, Python; CUDA programming experience

• Proficient in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, Warp

• Learn from and teach others to keep up to speed on the latest developments and tools in the field.

• Ability to work together in a tight-knit team environment.

Ways to stand out from the crowd:

• Interest and experience in the drug discovery domain

• Track record of code contributions that accelerate AI/ML bottlenecks

• Experience working with cutting edge methods in AI

With competitive salaries and a generous benefits package, NVIDIA is widely considered to be one of the technology industry's most desirable employers. We have some of the most forward-thinking and versatile people in the world working with us, and our engineering teams are growing fast in some of the most impactful fields of our generation: Deep Learning, Artificial Intelligence, and Autonomous Vehicles. If you're a creative engineer who enjoys autonomy and shares our passion for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.Applications for this job will be accepted at least until September 29, 2025.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.