Alle Stellen

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, a source-to-source compiler designed to unlock outstanding performance for PyTorch models on NVIDIA GPUs. This is an 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 open-source solutions that benefit the community.

Tasks

  • Lead the design, implementation, optimization, and maintenance of core compiler technologies that accelerate deep learning workloads.
  • Work collaboratively with engineers who built PyTorch for NVIDIA hardware to pioneer new features and stay at the forefront of framework development.
  • Conduct performance analysis of workloads running on thousands of GPUs to identify optimization opportunities for Thunder.
  • Collaborate with compiler, library, and systems teams, including experts behind nvFuser, TVM, XLA, and CUDA, to translate research into practical, high-impact open-source solutions.

Requirements

  • Bachelor's, Master's, or Ph.D. in Computer Science or a related technical 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 such as PyTorch or JAX, with an understanding of model construction and performance challenges.
  • Solid foundation in compiler concepts, including abstract syntax trees (ASTs), intermediate representations (e.g., SSA form), program analysis, and code generation.
  • Excellent communication and collaboration skills for effective work in a distributed, open-source environment.
  • Previous contributions to deep learning compiler projects (e.g., TVM, MLIR, IREE) or deep learning frameworks.
  • Deep expertise in the internals of PyTorch, particularly its compiler stack (TorchDynamo, TorchInductor).
  • Experience with JAX-like functional transformations in a compiler context.
  • Familiarity with parallel programming, distributed systems, and high-performance CUDA code.
  • Track record of impactful participation in open-source communities through code contributions, design discussions, or mentorship.

Benefits

  • Highly competitive salaries.
  • Extensive benefits package.
  • Work environment that promotes diversity, inclusion, and flexibility.
  • Supportive and empowering workplace as an equal opportunity employer.
  • Base salary for Poland: 292,500 PLN - 507,000 PLN, determined by location, experience, and comparable positions.
Bist du Teil dieses Unternehmens?

Dieses Unternehmensprofil wurde automatisch erstellt. Wenn du für NVIDIA Switzerland AG arbeitest, kannst du das Profil jetzt beanspruchen und verifizieren – kostenlos und in wenigen Minuten.

Verifizierte Profile erhalten ein Siegel und können ihre Seite, Stellen und Bewerbungen direkt verwalten.
Über uns
NVIDIA Switzerland AG ist die Schweizer Gesellschaft des internationalen Technologieunternehmens NVIDIA. Das Unternehmen ist in der Entwicklung, Vermarktung und Bereitstellung von Technologien, Produkten und Dienstleistungen in Bereichen wie künstliche Intelligenz, beschleunigtes Computing, Grafikprozessoren, Robotik, virtuelle Realität, Informatik und professionelle Visualisierung tätig. NVIDIA entwickelt weltweit Chips, Systeme, Software und Plattformen für Rechenzentren, Forschung, Gaming, Automobilindustrie, Industrieanwendungen, Gesundheitswesen und weitere technologieintensive Branchen. Die Schweizer Gesellschaft unterstützt die Aktivitäten von NVIDIA in der Schweiz und ist im Bereich IT-Dienstleistungen registriert.
Das Team

Work as part of a collaborative and innovative team, alongside engineers who built PyTorch for NVIDIA hardware, and in close collaboration with leading compiler, library, and systems teams.

Ähnliche Stellen