Staff AI Applications Engineer - Heterogeneous Compute Lead

Axelera AI; United States; Remote; USD 140,000 - 210,000; Full time

Benefits offered by this job

Rentenplan

Mitarbeitendenversicherung

Unternehmensanteile

Job summary

Axelera AI sucht einen Staff-KI-Anwendungsingenieur im Model-Enablement-Team. Du entwickelst die Model-Enablement-Plattform und planst, wie Modelle auf Metis-/Europa-AIPU, Host-CPU und Beschleunigern ausgeführt werden.

Du arbeitest an Frontend, Graph-Compiler, Runtime und der Anwendungspipeline, bringst Modelle durch Bring-up und verbesserst End-to-End-Leistung in enger Abstimmung mit Graph-Compiler- und Runtime-Teams.

Qualifications

• Mindestens 8 Jahre Berufserfahrung in Systemen, Compiler, Runtimes oder Performance.
• Praktische Erfahrung mit heterogenem Computing über CPU/GPU/DSP/NPU/FPGA.
• Fundierte Grundlagen in Rechnerarchitektur, Betriebssystemen und paralleler Programmierung.
• Vertrautheit mit ML-Graphrepräsentationen und Compilerkonzepten (Graph-Erfassung, Lowering, Operator-Fusion, Partitionierung, Speicherplanung).
• Sehr gute Kenntnisse in C++ und Python.
• Tägliche Nutzung von KI-Coding-Agenten bei realen Entwicklungsaufgaben.

Responsibilities

• Plattform Model-Enablement und Agent-Harness implementieren.
• Modellaufnahme analysieren und verteilen auf AIPU, Host-CPU und Beschleuniger.
• Heterogene Ausführung großer Modelle und Pipelines unterstützen.
• End-to-End-Leistung optimieren, Profiling durchführen.
• Kritische Modelle auf Metis/Europa analysieren und Verbesserungen ableiten.

Skills

Systemarchitektur

Heterogenes Computing

Betriebssysteme & Parallelisierung

ML-Graph-Rep & Compiler

C++

Python

KI-Coding-Agenten

Tools

MLIR

TVM

CUDA

ONNX-Runtime

Job description

Axelera AI sucht einen Staff-KI-Anwendungsingenieur im Model-Enablement-Team. Du entwickelst die Model-Enablement-Plattform und planst, wie Modelle auf Metis-/Europa-AIPU, Host-CPU und Beschleunigern ausgeführt werden.

Du arbeitest an Frontend, Graph-Compiler, Runtime und der Anwendungspipeline, bringst Modelle durch Bring-up und verbesserst End-to-End-Leistung in enger Abstimmung mit Graph-Compiler- und Runtime-Teams.