Senior Performance Engineer - AI for Science

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 Data Science @ 4 TensorFlow @ 6 Mathematics @ 7 Parallel Programming @ 7 PyTorch @ 6 CUDA @ 4 GPU @ 4

Details

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. Our cutting-edge GPU technology serves as the computational engine for groundbreaking research, enabling scientists to model complex biological systems, analyze and harmonize massive datasets, and accelerate the drug discovery process.

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. As part of our team, you'll work alongside researchers, developers, and experts in the field to tackle some of the most challenging and impactful problems of our time.

Responsibilities

  • Optimize the computational performance of AI innovations that address domain-relevant bottlenecks.
  • Design and implement computationally performant features for large-scale, CUDA-backed ML training and inference frameworks and data science tooling.
  • Develop and maintain an HPC software stack for generative machine learning models in digital biology and beyond.
  • Collaborate with multiple HPC, AI infrastructure, and research teams.
  • Drive testing and maintenance of algorithms and software modules.

Requirements

  • 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++ and Python; CUDA programming experience.
  • Proficient in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, and Warp.
  • Experience developing and maintaining HPC software stacks and working with large-scale ML training and inference workloads.
  • Ability to learn from and teach others; strong collaboration and teamwork skills.

Ways to stand out

  • 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 & Benefits

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5. You will also be eligible for equity and benefits.

Additional information

  • Location: Santa Clara, CA, United States.
  • Employment type: Full time.
  • Applications for this job will be accepted at least until September 29, 2025.
  • NVIDIA is an equal opportunity employer and committed to fostering a diverse work environment.