Senior Software Engineer, JAX — NVIDIA (ufficio Zurich)
CHF 101'500 - 154'000
NVIDIA (ufficio Zurich) · Zürich (ZH)
- Location
- Zürich
- Contract
- full-time
- Posted
- 32 days ago
SalaryCHF 101'500 - 154'000
Role overview
NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code.
Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX.
To achieve this and provide an excellent user experience, you have strong programming skills and excellent system design skills.
- NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX, a tool that can differentiate between Python control code and NumPy code.
- Our goal is to deliver a polished, fast, modular and coordinated platform to handle data, training and analysis for a wide range of deep learning solutions using JAX.
Application process
- Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products. What we need to see:
- 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.
Company and context
- Coordinates effectively across organizational boundaries and geographies. Ways to stand out from the crowd:
- Understanding of JAX, Autograd, tracing, code generation and DSL compilers and their design.
- Understanding of deep learning training in distributed contexts: multi-GPU, multi-node, synchronous vs asynchronous.
- Background with software shipping cycles (dev, deploy, release, CI).
- Experience building distributed systems and services at large scale.
- Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
- For Poland: The base salary range is 292,500 PLN
- 507,000 PLN for Level 4, and 375,000 PLN
- 650,000 PLN for Level 5.
Additional details
- Come join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world! What you will be doing:
- Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products. What we need to see:
- Coordinates effectively across organizational boundaries and geographies. Ways to stand out from the crowd:
Notes and original content
- Come join us to build the next generation of tooling that will advance the numerical computing and machine learning research around the world!
- What you will be doing:
- Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products.
- What we need to see:
- Coordinates effectively across organizational boundaries and geographies.
- Ways to stand out from the crowd:
