Senior Deep Learning Compiler Engineer - PyTorch — NVIDIA (ufficio Zurich)
CHF 101'500 - 154'000
NVIDIA (ufficio Zurich) · Zürich (ZH)
- Location
- Zürich
- Contract
- full-time
- Posted
- 31 days ago
SalaryCHF 101'500 - 154'000
Role overview
Join us at the forefront of AI compiler technology and help shape the future of accelerated computing.
NVIDIA is seeking passionate engineers to build the next generation of tools used by AI developers and researchers worldwide.
Our team is developing Thunder , an ambitious, source-to-source compiler built to unlock outstanding performance for PyTorch models on NVIDIA GPUs.
- Join us at the forefront of AI compiler technology and help shape the future of accelerated computing.
- NVIDIA is seeking passionate engineers to build the next generation of tools used by AI developers and researchers worldwide.
- This is a unique opportunity to contribute to a project that enhances the PyTorch ecosystem, working with modern compiler stacks like PyTorch 2.0's TorchDynamo and TorchInductor to create powerful, open-source solutions that benefit the entire community.
What the company offers
- This is a unique opportunity to contribute to a project that enhances the PyTorch ecosystem, working with modern compiler stacks like PyTorch 2.0's TorchDynamo and TorchInductor to create powerful, open-source solutions that benefit the entire community.
Application process
- If you are driven to solve complex problems and want to make a foundational impact on the AI ecosystem, apply to join our collaborative and innovative team. What you'll be doing:
- Experience with JAX-like functional transformations and their application in a compiler context.
- Familiarity with parallel programming, distributed systems, and writing high-performance CUDA code.
- A track record of impactful participation in open-source communities, such as through code contributions, design discussions, or mentorship.
- NVIDIA is at the forefront of breakthroughs in Artificial Intelligence, High-Performance Computing, and Visualization.
- Our teams are composed of driven, innovative professionals dedicated to pushing the boundaries of technology.
- We offer highly competitive salaries, an extensive benefits package, and a work environment that promotes diversity, inclusion, and flexibility.
- As an equal opportunity employer, we are committed to fostering a supportive and empowering workplace for all.
Additional details
- If you are driven to solve complex problems and want to make a foundational impact on the AI ecosystem, apply to join our collaborative and innovative team. What you'll be doing:
- Furthermore, you will be part of a vibrant ecosystem, working closely with leading compiler, library, and systems teams—including experts behind nvFuser, TVM, XLA, and CUDA—to translate the latest research into practical, high-impact solutions for the open-source community. What we need to see:
- Excellent communication and collaboration skills, essential for working effectively in a distributed, open-source environment. Ways to stand out from the crowd:
- Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. For Poland: The base sal
Notes and original content
- If you are driven to solve complex problems and want to make a foundational impact on the AI ecosystem, apply to join our collaborative and innovative team.
- What you'll be doing:
- Furthermore, you will be part of a vibrant ecosystem, working closely with leading compiler, library, and systems teams—including experts behind nvFuser, TVM, XLA, and CUDA—to translate the latest research into practical, high-impact solutions for the open-source community.
- What we need to see:
- Excellent communication and collaboration skills, essential for working effectively in a distributed, open-source environment.
- Ways to stand out from the crowd:
- Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
- For Poland: The base sal
