Senior Ml Framework Performance Engineer - Ai For Science At Scale

Invidia

CA, United States
Base: 184,000 usd - 287,500 usd; bonus/equity: not...
Performance engineering in ai
Modern machine learning frameworks
Deep understanding of compute bottlenecks
NVIDIA is seeking a Senior Machine Learning Performance Engineer to advance AI for Science

Job Summary

  • NVIDIA is seeking a Senior Machine Learning Performance Engineer to advance AI for Science.
  • The role involves designing performance evaluation frameworks for ML models in scientific discovery.
  • NVIDIA offers competitive salaries and a comprehensive benefits package.

Matching Summary

NVIDIA is seeking a Senior Machine Learning Performance Engineer to advance AI for Science.

Salary

Base: 184,000 USD - 287,500 USD; Bonus/Equity: Not specified; Benefits: Not specified

Skills & Requirements

Must-have

  • performance engineering in AI
  • modern machine learning frameworks
  • deep understanding of compute bottlenecks

Nice-to-have

  • contributor to scientific codebase
  • experience with CUDA/Triton programming
  • strong communication skills

Key Requirements

  • Advanced degree in quantitative field
  • 5+ years of relevant experience
  • expertise in PyTorch and JAX

Work Rights

Not specified

Tailored Resume

Cover Letter