Logo
NVIDIA

Senior Performance Engineer - AI for Science

NVIDIA, New York, New York, us, 10261

Save Job

Senior Performance Engineer - AI for Science

Join NVIDIA as a 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. 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. 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 NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. 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. The base salary range is $184,000 - $287,500 for Level 4, and $224,000 - $356,500 for Level 5. You will also be eligible for equity and benefits.

#J-18808-Ljbffr