Senior HPC Performance Engineer - AI for Science at Scale

at Nvidia
📍 Santa Clara, United States
USD 184,000-356,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 7 Algorithms @ 4 Machine Learning @ 4 TensorFlow @ 6 Leadership @ 4 Communication @ 7 Mathematics @ 7 Parallel Programming @ 7 Technical Leadership @ 4 PyTorch @ 6 CUDA @ 4 GPU @ 4

Details

NVIDIA is seeking a Senior HPC Performance Engineer to join teams building next-generation scientific machine learning (ML) frameworks. The role focuses on accelerating AI for Science through high-performance computing (HPC), CUDA-backed ML training frameworks, and advances in algorithmic and software stack performance for domains such as digital biology.

Responsibilities

  • Design and implement computationally performant features for large-scale, CUDA-backed ML training frameworks using low-level acceleration and scaling strategies such as GPU porting, data structure innovations, and distributed learning technologies.
  • Optimize computational performance of a wide range of business-critical ML models via the accelerated hardware and software stack, as well as algorithmic improvements.
  • Develop and maintain the HPC software stack for generative machine learning models in digital biology and other scientific domains.
  • Collaborate with multiple HPC, AI infrastructure, and research teams.
  • Develop tools to assist data processing, data quality control, algorithm development, and algorithm 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, packaging, and launching software products.
  • Deep understanding of parallel programming in C++ and Python; ideally CUDA and Triton programming experience.
  • Fluent in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, and Warp.
  • Experience applying HPC solutions to research problems, ideally for biology, chemistry, or material science applications.
  • Recognized for technical leadership contributions, capable of self-direction, and able to learn from and teach others.
  • Strong communication skills, organized, self-motivated, and an excellent teammate.

Ways to stand out

  • Contributor to major scientific AI for Science codebases.
  • Familiarity with pioneering language and geometric models used in AI for Science applications in biology, chemistry, or material science.

Compensation & Benefits

  • Base salary ranges by level:
    • Level 4: 184,000 USD - 287,500 USD
    • Level 5: 224,000 USD - 356,500 USD
  • You will also be eligible for equity and benefits (link provided in the original posting).

Other details

  • Applications for this job will be accepted at least until October 4, 2025.
  • NVIDIA is an equal opportunity employer and values diversity.