Senior Software Engineer, Jax

Invidia

Multiple Locations
Base: 184,000 usd - 287,500 usd for level 4; 224,0...
C/c++ and python programming
Machine learning framework internals
Cpu and gpu architecture knowledge
NVIDIA is hiring senior engineers to develop its AI platform and optimize performance in deep learning frameworks using JAX

Job Summary

  • NVIDIA is hiring senior engineers to develop its AI platform and optimize performance in deep learning frameworks using JAX.
  • The role involves designing and implementing core JAX components, collaborating with AI researchers, and building tools to improve AI system development efficiency.
  • Employees are eligible for competitive base salaries, equity, and benefits, and the company fosters a diverse and inclusive work environment.

Matching Summary

NVIDIA is hiring senior engineers to develop its AI platform and optimize performance in deep learning frameworks using JAX.

Salary

Base: 184,000 USD - 287,500 USD for Level 4; 224,000 USD - 356,500 USD for Level 5; Bonus/Equity: Eligible for equity; Benefits: Eligible for benefits

Skills & Requirements

Must-have

  • C/C++ and Python programming
  • Machine learning framework internals
  • CPU and GPU architecture knowledge
  • Modular software design
  • Performance optimization in deep learning
  • System design skills

Nice-to-have

  • Understanding of JAX and Autograd
  • Experience with distributed deep learning training
  • Software shipping cycles experience
  • Building distributed systems at scale
  • Excellent communication and planning skills

Key Requirements

  • BS in Computer Science or related field or equivalent experience
  • 6+ years relevant experience
  • Proven customer-facing solution development
  • Ability to work with multi-functional teams
  • Excellent verbal and written communication skills

Work Rights

Not specified

Tailored Resume

Cover Letter