Senior Deep Learning Compiler Engineer - PyTorch job opportunity at NVIDIA.



DateMore Than 30 Days Ago bot
NVIDIA Senior Deep Learning Compiler Engineer - PyTorch
Experience: 8-years
Pattern: full-time
apply Apply Now
Salary:
Status:
Copy Link Report
degreeBachelor's (B.A.)
loacation Germany, Berlin, Germany
loacation Germany, Berli..........Germany

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. 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. 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: As a key member of our team, you will be contributing directly to the future of accelerated AI. Your role will be dynamic and deeply technical, placing you at the center of compiler innovation. You will lead the design, implementation, optimization, and maintenance of the core compiler technologies that accelerate massive deep learning workloads. This is a highly collaborative role where you'll work alongside the very engineers who built PyTorch for NVIDIA hardware, helping to pioneer new features and stay at the forefront of framework development. You'll dive deep into performance analysis, scrutinizing workloads running on thousands of GPUs to find optimization opportunities that will shape the future design of Thunder. 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: We are looking for engineers who are excited about building powerful, user-centric tools and are comfortable working in a fast-paced, collaborative environment. Here are some of the expertise we would like to see: A Bachelor's, Master's, or Ph.D. in Computer Science or a related technical field (or equivalent experience). 8+ years of relevant work experience A strong command of Python and experience building complex, well-tested software systems. Hands-on experience with deep learning frameworks like PyTorch or JAX. You understand how models are built and where the performance challenges lie. A solid foundation in compiler concepts such as abstract syntax trees (ASTs), intermediate representations (e.g., SSA form), program analysis, and code generation. Excellent communication and collaboration skills, essential for working effectively in a distributed, open-source environment. Ways to stand out from the crowd: Previous contributions to deep learning compiler projects (e.g., TVM, MLIR, IREE) or deep learning frameworks themselves. Deep expertise in the internals of PyTorch, particularly its compiler stack (TorchDynamo, TorchInductor). 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. 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.




Other Ai Matches

Senior Software Engineer – Simulation and Virtualization Applicants are expected to have a solid experience in handling Job related tasks
remote-jobserver Remote
Senior Deep Learning Performance Architect Applicants are expected to have a solid experience in handling Job related tasks
Senior System Software Architect, AI Networking Applicants are expected to have a solid experience in handling AI Networking related tasks
Senior Software Development Engineer in Test, Confidential Computing - SDET Applicants are expected to have a solid experience in handling Confidential Computing - SDET related tasks
Senior Developer Relations Manager Applicants are expected to have a solid experience in handling Job related tasks
Senior Technical Program Manager, Data Center - Engineering Operations Applicants are expected to have a solid experience in handling Data Center - Engineering Operations related tasks
AI Factory Digital Twin Engineer Applicants are expected to have a solid experience in handling Job related tasks
Agentic AI Solution Engineering Intern - Summer 2026 Applicants are expected to have a solid experience in handling Job related tasks
remote-jobserver Remote
Technical Marketing Engineer Applicants are expected to have a solid experience in handling Job related tasks
Software Design Engineer - SONiC Group Applicants are expected to have a solid experience in handling Job related tasks
Senior System Software Engineer - Tegra MODS Team Applicants are expected to have a solid experience in handling Job related tasks
Senior Product System Engineer Applicants are expected to have a solid experience in handling Job related tasks
Senior Technical Program Manager - Automotive Vehicles Applicants are expected to have a solid experience in handling Job related tasks
remote-jobserver Remote
Developer Relations Manager -Nordics Applicants are expected to have a solid experience in handling Job related tasks
ASIC Design Engineer Applicants are expected to have a solid experience in handling Job related tasks
Senior ASIC Verification Engineer, Coherent High Speed Interconnect Applicants are expected to have a solid experience in handling Coherent High Speed Interconnect related tasks
Senior Software Engineer - VLM Microservices for Neural Reconstruction Applicants are expected to have a solid experience in handling Job related tasks
Senior Counsel, Commercial Legal Applicants are expected to have a solid experience in handling Commercial Legal related tasks
Senior CAD Engineer - DFX Software Applicants are expected to have a solid experience in handling Job related tasks
Solutions Architect, AI Hyperscalers Applicants are expected to have a solid experience in handling AI Hyperscalers related tasks
System Debug Lead Engineer Applicants are expected to have a solid experience in handling Job related tasks
Senior Validation Architect, NVLink Applicants are expected to have a solid experience in handling NVLink related tasks
Senior System Software Engineer - AV Platform Applicants are expected to have a solid experience in handling Job related tasks