Senior HPC Performance Engineer - AI for Science

September 15

Apply Now

Receive Emails with Similar Jobs

NVIDIA

WebsiteLinkedInAll Job Openings

GPU-accelerated computing • artificial intelligence • deep learning • virtual reality • gaming

10,000+

Description

• Design and implement computationally performant features for large scale, CUDA-backed ML training frameworks • Optimize computational performance of business-critical ML models via accelerated hardware and software stack • Develop and maintain HPC software stack for generative machine learning models in digital biology and beyond • Collaborate with multiple HPC, AI infrastructure, and research teams • Develop tools to assist data processing, quality control, and algorithm development and testing • Drive the testing and maintenance of algorithms and software modules

Requirements

• Advanced degree in a quantitative field such as Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics, or equivalent experience • 8+ years of relevant experience • Consistent 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; ideally CUDA programming experience • Fluent in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, Warp • Experience with HPC solutions to research problems, ideally for biology, chemistry or material science applications • Recognized for technical leadership contributions, capable of self-direction, and ability to learn from and teach others • Strong communication skills, organized and self-motivated, and able to work well in a team

Benefits

• Highly competitive salaries • Comprehensive benefits package