Senior Ml Framework Performance Engineer - Ai For Science At Scale

NVIDIA

Base: 184,000 usd - 287,500 usd; bonus/equity: eli...
Not specified
Performance engineering for large scale ai models
Expertise in pytorch, jax, warp frameworks
Model parallelism and distributed learning
NVIDIA is seeking a Senior Machine Learning Performance Engineer to support the development of advanced scientific machine learning frameworks, particularly in digital biology. The role emphasizes performance evaluation, identifying computational bottlenecks, and collaborating with various teams to enhance AI capabilities for scientific applications

Job Summary

  • NVIDIA is advancing AI for Science by building next generation scientific machine learning frameworks through collaborations with industry and academia.
  • The role involves designing performance evaluation frameworks, identifying bottlenecks, and driving testing and maintenance of AI for Science algorithms and software.
  • NVIDIA offers highly competitive salaries, equity, comprehensive benefits, and a dynamic environment working on cutting-edge fields like Quantum Computing and Autonomous Vehicles.

Matching Summary

Match Score: 85

NVIDIA is seeking a Senior Machine Learning Performance Engineer to support the development of advanced scientific machine learning frameworks, particularly in digital biology. The role emphasizes performance evaluation, identifying computational bottlenecks, and collaborating with various teams to enhance AI capabilities for scientific applications.

Salary

Base: 184,000 USD - 287,500 USD; Bonus/Equity: Eligible for equity; Benefits: Comprehensive benefits package

Skills & Requirements

Must-have

  • Performance engineering for large scale AI models
  • Expertise in PyTorch, JAX, Warp frameworks
  • Model parallelism and distributed learning
  • Machine learning in scientific discovery
  • Software design and product launching

Nice-to-have

  • Contributor to scientific atomistic modeling codebase
  • Experience with CUDA/Triton programming
  • Strong communication and teamwork skills
  • Self-motivated and organized
  • Technical leadership and mentoring

Key Requirements

  • Advanced degree in quantitative field or equivalent experience
  • 5+ years relevant experience in AI model performance engineering
  • Deep understanding of compute bottlenecks and parallelism paradigms
  • Up-to-date knowledge of ML research in atomistic modeling
  • Experience with ML research software or atomistic simulation tools

Work Rights

Not specified

Tailored Resume

Cover Letter