Senior Deep Learning Compiler Engineer - PyTorch

Schliesse dich uns an der Spitze der KI-Compiler-Technologie an und hilf, die Zukunft des beschleunigten Rechnens zu gestalten. NVIDIA sucht leidenschaftliche Ingenieure, um die nächste Generation von Werkzeugen zu entwickeln, die von KI-Entwicklern und Forschern weltweit genutzt werden. Unser Team entwickelt Thunder, einen ehrgeizigen Source-to-Source-Compiler, der herausragende Leistung für PyTorch-Modelle auf NVIDIA-GPUs freischaltet. Dies ist eine einzigartige Gelegenheit, zu einem Projekt beizutragen, das das PyTorch-Ökosystem verbessert, indem moderne Compiler-Stacks wie PyTorch 2.0’s TorchDynamo und TorchInductor verwendet werden, um leistungsstarke, Open-Source-Lösungen zu schaffen, die der gesamten Gemeinschaft zugutekommen.

Wenn du motiviert bist, komplexe Probleme zu lösen und einen grundlegenden Einfluss auf das KI-Ökosystem ausüben möchtest, bewirb dich, um unserem kollaborativen und innovativen Team beizutreten.

Was du tun wirst:

• Als Schlüsselmitglied unseres Teams trägst du direkt zur Zukunft der beschleunigten KI bei.
• Deine Rolle wird dynamisch und tief technisch sein, was dich ins Zentrum der Compiler-Innovation stellt.
• Du wirst das Design, die Implementierung, Optimierung und Wartung der Kerncompiler-Technologien leiten, die massive Deep-Learning-Workloads beschleunigen.
• Dies ist eine hochgradig kollaborative Rolle, in der du eng mit den Ingenieuren zusammenarbeitest, die PyTorch für NVIDIA-Hardware entwickelt haben.
• Du wirst tief in die Leistungsanalyse eintauchen und Workloads, die auf Tausenden von GPUs laufen, genau unter die Lupe nehmen, um Optimierungsmöglichkeiten zu finden.
• Darüber hinaus wirst du Teil eines lebendigen Ökosystems sein und eng mit führenden Compiler-, Bibliotheks- und Systemteams zusammenarbeiten.

Was wir sehen müssen:

• Ein Bachelor-, Master- oder Doktortitel in Informatik oder einem verwandten technischen Bereich (oder gleichwertige Erfahrung).
• 8+ Jahre relevante Berufserfahrung.
• Ein starkes Verständnis von Python und Erfahrung im Aufbau komplexer, gut getesteter Softwaresysteme.
• Praktische Erfahrung mit Deep-Learning-Frameworks wie PyTorch oder JAX.
• Ein solides Fundament in Compiler-Konzepten wie abstrakten Syntaxbäumen (ASTs), Zwischenrepräsentationen, Programm-Analyse und Code-Generierung.
• Ausgezeichnete Kommunikations- und Kollaborationsfähigkeiten, die für die effektive Arbeit in einer verteilten, Open-Source-Umgebung unerlässlich sind.

Wie du dich von der Masse abheben kannst:

• Frühere Beiträge zu Deep-Learning-Compiler-Projekten oder Deep-Learning-Frameworks selbst.
• Tiefe Expertise in den Interna von PyTorch, insbesondere in seinem Compiler-Stack.
• Erfahrung mit JAX-ähnlichen funktionalen Transformationen und deren Anwendung im Compiler-Kontext.
• Vertrautheit mit paralleler Programmierung, verteilten Systemen und dem Schreiben von Hochleistungs-CUDA-Code.
• Eine Erfolgsbilanz von wirkungsvoller Teilnahme an Open-Source-Communities.

NVIDIA steht an der Spitze von Durchbrüchen in der Künstlichen Intelligenz, Hochleistungsrechnen und Visualisierung. Unsere Teams bestehen aus engagierten, innovativen Fachleuten, die sich dafür einsetzen, die Grenzen der Technologie zu erweitern. Wir bieten wettbewerbsfähige Gehälter, ein umfangreiches Leistungspaket und ein Arbeitsumfeld, das Vielfalt, Inklusion und Flexibilität fördert.