Senior Software Engineer, Jax

NVIDIA

Base: 292,500 pln - 507,000 pln for level 4; bonus...
C/c++ and python programming
Experience with machine learning frameworks
Proven technical foundation in cpu and gpu architectures
NVIDIA is hiring senior engineers to develop its AI platform with a focus on performance optimizations in deep learning frameworks using JAX

Job Summary

  • NVIDIA is hiring senior engineers to develop its AI platform with a focus on performance optimizations in deep learning frameworks using JAX.
  • The role involves designing and implementing JAX core components and collaborating with AI researchers to build efficient models.
  • Join us to advance numerical computing and machine learning research globally.

Matching Summary

NVIDIA is hiring senior engineers to develop its AI platform with a focus on performance optimizations in deep learning frameworks using JAX.

Salary

Base: 292,500 PLN - 507,000 PLN for Level 4; Bonus/Equity: Not specified; Benefits: Not specified

Skills & Requirements

Must-have

  • C/C++ and Python programming
  • Experience with machine learning frameworks
  • Proven technical foundation in CPU and GPU architectures

Nice-to-have

  • Understanding of JAX and Autograd
  • Experience building distributed systems
  • Background with software shipping cycles

Key Requirements

  • BS in Computer Science or related field
  • 6+ years relevant experience
  • Ability to work with multi-functional teams

Work Rights

Not specified

Tailored Resume

Cover Letter