← All roles

Senior Deep Learning Compiler Engineer - PyTorch

NVIDIA· Engineer, Tech SW
Apply on company site
Hybrid (Berlin, Germany), Hybrid (Amsterdam, Netherlands), Hybrid (Warsaw, Poland)PLN 292.5k - PLN 507kfull-timePosted 7mo ago

About the role

Senior role contributing to the Thunder project, a source-to-source compiler to accelerate PyTorch models on NVIDIA GPUs. Focus on design, implementation, optimization, and maintenance of core compiler technologies, collaborating with teams across nvFuser, TVM, XLA, and CUDA to deliver open-source solutions for the PyTorch ecosystem.

Responsibilities

  • Lead design, implementation, optimization, and maintenance of core compiler technologies that accelerate massive deep learning workloads.
  • Dive deep into performance analysis of workloads running on thousands of GPUs to identify optimization opportunities.
  • Collaborate with compiler, library, and systems teams (nvFuser, TVM, XLA, CUDA) to translate research into practical, high-impact solutions for the open-source community.

Qualifications

  • Bachelor's, Master's, or Ph.D. in Computer Science or related field (or equivalent experience)
  • 8+ years of relevant work experience
  • Strong command of Python and experience building complex, well-tested software systems
  • Hands-on experience with deep learning frameworks like PyTorch or JAX
  • Solid foundation in compiler concepts (ASTs, SSA, program analysis, code generation)
  • Excellent communication and collaboration skills for distributed, open-source environments

Skills

Required:PythonDeep Learning FrameworksCompiler Concepts

Benefits

  • Highly competitive salaries
  • Extensive benefits package
  • Work environment that promotes diversity, inclusion, and flexibility
Last updated Oct 23, 2026. Interested? Apply on the company site.

About NVIDIA

World Leader in AI Computing

NVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.

AIautonomous vehiclesgaminggpugraphics cardhigh performance computingliquid-ai-recruiting-targetllm
Company profileWebsite

More open roles at NVIDIA