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 Mentoring @ 4 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 frameworks for AI for Science (starting with digital biology). The role focuses on designing and implementing computationally performant features for large-scale, CUDA-backed ML training frameworks, optimizing model performance across hardware and software stacks, and developing/maintaining HPC software stacks for generative ML models.

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 accelerated hardware and software stack improvements and algorithmic enhancements.
  • Develop and maintain the HPC software stack for generative machine learning models in digital biology and other 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 testing and maintenance of algorithms and software modules.

Requirements

  • Advanced degree in a quantitative field (e.g., Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics) or equivalent experience.
  • 8+ years of relevant experience.
  • Proven track record in performance engineering, software design, building/packaging, and launching software products.
  • Deep understanding of parallel programming in C++ and Python; CUDA programming experience is preferred/ideal.
  • Proficient with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, and Warp.
  • Experience applying HPC solutions to research problems, ideally in biology, chemistry, or materials science.
  • Recognized for technical leadership contributions; capable of self-direction and mentoring others.
  • Strong communication skills, organization, self-motivation, and team collaboration.

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, and materials science.

Compensation & Logistics

  • 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.
  • Location: US — CA — Santa Clara.
  • Employment type: Full time.
  • Applications accepted at least until August 14, 2025.

About NVIDIA

NVIDIA emphasizes competitive salaries, a comprehensive benefits package, and is an equal opportunity employer valuing diversity. The company’s engineering teams work across state-of-the-art fields including Digital Biology, Artificial Intelligence, and Autonomous Vehicles.