Senior Software Engineer, JAX
NVIDIASwitzerland • United Kingdom • Poland • Germany • France
Posted April 17, 2026
Full Time
About the Company
Join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world!
Job Description
NVIDIA is hiring senior engineers to develop its AI platform using JAX and improve performance optimizations in deep learning frameworks.
Requirements
- BS in Computer Science or Computer Engineering or related field (or equivalent experience)
- 6+ years relevant experience
- C/C++ and Python programming
- Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.)
- Proven ability developing customer-facing solutions, balancing feature requests and bugs
- Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design
- Highly motivated with excellent verbal and written communication skills
- Ability to work successfully with multi-functional teams, principles and architects
Benefits
- Base salary determined by location, experience, and pay of employees in similar positions
- Determined by location, experience, and pay of employees in similar positions