Senior Performance Engineer - AI for Science

Nvidia • Santa Clara, CA

91d • $184,000 - $287,500

About The Position

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.

Requirements

• 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.
• Ability to work together in a tight-knit team environment.

Nice To Haves

• 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.

Responsibilities

• 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.

Benefits

• Competitive salaries.
• Generous benefits package.
• Equity eligibility.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Master's degree