NVIDIA
Senior Performance Engineer - AI for Science
NVIDIA, California, Missouri, United States, 65018
Overview
Join to apply for the Senior Performance Engineer - AI for Science role at NVIDIA. NVIDIA has a long history of accelerated computing and is applying this legacy to digital biology and drug discovery. Our GPU technology enables researchers to model biological systems, analyze large datasets, and accelerate the drug discovery process. We are looking for a Senior Performance Engineer who shares our vision for transforming healthcare. This is a fast-paced, collaborative environment where you will help develop tools that redefine what is possible in drug discovery, working with researchers, developers, and field experts on impactful problems. What You’ll Be Doing
Optimizing the computational performance of the latest AI innovations to address domain-relevant bottlenecks. Designing and implementing 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. Ability to learn from and teach others to keep up to speed on the latest developments and tools in the field. Ability to work well 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. Compensation and Benefits
With competitive salaries and a generous benefits package, NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer. Our salary ranges are determined by location, experience, and internal benchmarks. Applicable equity and benefits are included. Applications for this job will be accepted at least until September 29, 2025. EEO Statement
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. We do not discriminate 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. JR2004844
#J-18808-Ljbffr
Join to apply for the Senior Performance Engineer - AI for Science role at NVIDIA. NVIDIA has a long history of accelerated computing and is applying this legacy to digital biology and drug discovery. Our GPU technology enables researchers to model biological systems, analyze large datasets, and accelerate the drug discovery process. We are looking for a Senior Performance Engineer who shares our vision for transforming healthcare. This is a fast-paced, collaborative environment where you will help develop tools that redefine what is possible in drug discovery, working with researchers, developers, and field experts on impactful problems. What You’ll Be Doing
Optimizing the computational performance of the latest AI innovations to address domain-relevant bottlenecks. Designing and implementing 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. Ability to learn from and teach others to keep up to speed on the latest developments and tools in the field. Ability to work well 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. Compensation and Benefits
With competitive salaries and a generous benefits package, NVIDIA is committed to fostering a diverse work environment and is an equal opportunity employer. Our salary ranges are determined by location, experience, and internal benchmarks. Applicable equity and benefits are included. Applications for this job will be accepted at least until September 29, 2025. EEO Statement
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. We do not discriminate 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. JR2004844
#J-18808-Ljbffr