Senior Compiler Engineer – AI Accelerators (LLVM/MLIR)

Bei AMD verändern Sie alles. Unsere Mission ist es, großartige Produkte zu entwickeln, die die nächste Generation von Computererlebnissen beschleunigen – von KI und Rechenzentren bis hin zu PCs, Gaming und eingebetteten Systemen. Wir glauben, dass echter Fortschritt aus mutigen Ideen, menschlicher Kreativität und einer gemeinsamen Leidenschaft entsteht, etwas Außergewöhnliches zu schaffen.

Wir suchen einen praktischen Compiler-Ingenieur, der hilft, den Compiler-Stack von AMD für Neural Processing Units (NPUs) aufzubauen und zu optimieren. Diese Rolle konzentriert sich auf die Entwicklung von Frontend- und Mid-End-Compilern für KI-Arbeitslasten, einschließlich Graphtransformationen, Operatorfusion, Tiling, speichergestützte Optimierungen und effiziente Zuordnung zur Beschleunigerhardware.

Der erfolgreiche Kandidat bringt starke Software-Engineering-Grundlagen, praktische Compiler-Erfahrung und ein solides Verständnis der Interaktion zwischen Hardware und Software mit. Wir begrüßen Bewerber mit Erfahrung in architekturbezogener Softwareoptimierung, einschließlich der Kompilierung für Beschleuniger, ASICs, FPGAs, GPUs, NPUs oder andere spezialisierte Hardware.

Hauptverantwortlichkeiten:

• Entwurf, Entwicklung und Wartung von Compiler-Komponenten, die auf AMD AI-Beschleuniger abzielen, mit Schwerpunkt auf Frontend- und Mid-End-Transformationen.
• Implementierung und Verbesserung von Compiler-Optimierungen, einschließlich Operatorfusion, Tiling, Speicherlayout-Transformationen, Planung und Unterstützung der Codegenerierung für KI-Arbeitslasten.
• Verwendung von C++ und Python zur Entwicklung von Compiler-Pässen, Testinfrastruktur und Entwicklungstools auf Basis von LLVM/MLIR und verwandten Open-Source-Technologien.
• Analyse der Modell- und Benchmark-Leistung, Debugging von Compiler-Problemen und Identifizierung effizienter Zuordnungen von Arbeitslasten zur Hardware.
• Zusammenarbeit mit Hardwarearchitekten, Runtime-Teams und Teams für maschinelles Lernen zur Verbesserung der Compiler-Qualität, -Korrektheit und -Leistung.
• Beitrag zu Build-, Test- und Entwicklungsinfrastrukturen zur Verbesserung der Robustheit, Skalierbarkeit und Ingenieureffizienz.
• Übernahme technisch bedeutender Projekte bei effektiver Zusammenarbeit über verteilte Teams hinweg.

Qualifikationen:

• Abschluss in Informatik, Computertechnik, Elektrotechnik oder einem verwandten technischen Bereich mit 8+ Jahren Erfahrung oder Master-Abschluss mit 6+ Jahren oder PhD mit 3+ Jahren Erfahrung in Compiler-Engineering, Systemsoftware oder leistungsorientierter Softwareentwicklung.
• Starke C++-Programmierkenntnisse und Kenntnisse in Python.
• Verständnis der Compiler-Grundlagen, einschließlich Zwischenrepräsentationen, Optimierungspässe, Codegenerierung und Debugging.
• Erfahrung mit LLVM, MLIR, TVM oder ähnlichen Compiler-Frameworks.
• Erfahrung in der Implementierung von Compiler-Pässen innerhalb von LLVM/MLIR oder vergleichbarer Compiler-Infrastruktur.
• Kenntnisse über Softwareoptimierungen in Bezug auf Hardwarearchitekturmerkmale, einschließlich Speicherhierarchie, Datenbewegung, Threading, Vektorisierung oder Ausführungsmodelle für Beschleuniger.
• Erfahrung mit der Kompilierung oder Bereitstellung von KI/ML-Modellen oder mit der Kompilierung für spezialisierte Hardware wie GPUs, NPUs, DSPs, ASICs oder FPGAs.
• Starke Problemlösungs-, Debugging- und Kommunikationsfähigkeiten.

Bevorzugte Qualifikationen:

• Erfahrung mit ONNX, PyTorch oder anderen Frameworks für maschinelles Lernen und deren Compiler-/Runtime-Schnittstellen.
• Erfahrung mit graphbasierten Transformationen, domänenspezifischen Zwischenrepräsentationen (IRs) oder ML-fokussierten Optimierungen.
• Beiträge zu Open-Source-Compiler- oder KI-Infrastrukturprojekten.
• Erfahrung in Linux-basierten Entwicklungsumgebungen und mit modernen Software-Engineering-Tools wie Git, CI-Systemen, Debuggern und Profiler.
• Vertrautheit mit Software-Stacks für KI-Beschleuniger oder eingebettete/hochleistungsfähige Computergeräte.

Bildung:

• Bachelor-, Master- oder PhD-Abschluss in Informatik, Computertechnik, Elektrotechnik oder einem verwandten technischen Bereich.

AMD bietet eine Vielzahl von Vorteilen und fördert Chancengleichheit in der Beschäftigung.