Staff AI App Engineer — Heterogeneous Compute (Remote)

Axelera AI; France; Remote; EUR 120,000 - 180,000; Full time

Benefits offered by this job

a pension plan

extensive employee insurances

the option to get company shares

Job summary

Axelera AI is seeking a Staff AI Application Engineer to build the model enablement platform and decide how each model runs across the Metis or Europa AIPU, the host CPU and accelerators. The role covers the frontend, graph compiler, runtime and application pipeline, with model bring-up spanning frontend and graph sides.

You will work with AI coding agents in daily tasks and collaborate with field engineers, tech leads and managers who want to stay hands-on.

Qualifications

• 8+ years of engineering experience in systems, compilers, runtimes or performance.
• Hands-on experience with heterogeneous computing across CPU, GPU, DSP, NPU or FPGA.
• Solid fundamentals in computer architecture, operating systems and parallel programming.
• Familiarity with ML graph representations and compiler concepts: graph capture (torch.export, FX, ONNX), lowering, operator fusion, partitioning, memory planning.
• Strong C++ and Python.
• Daily use of AI coding agents on real engineering work, with the judgment to review and test their output.

Responsibilities

• Model enablement platform and agent harness: Build the platform that engineers and AI agents use to bring up models: reusable tools to partition, compile, run and profile; environments with access to real hardware; closed-loop runs that apply a change, measure it against the reference and repeat; and verification integrated into CI.
• Model ingestion: Own how a PyTorch or ONNX model is captured, analysed and split across the AIPU's matrix and RISC‑V vector units, the host CPU and the host's accelerators (GPU, NPU, video decoder), with placement tags that users can read and override.
• Heterogeneous execution: Extend that split to large models, cascades and multi-model pipelines, for example with expert offloading and caching across device and host memory.
• End-to-end performance: With the application team (pipeline builder) and the runtime team, bring each model to its best end-to-end performance: overlap of host and device work, zero-copy buffers, batching, multi-core scheduling, and models with several inputs.
• Critical models: Study the models that matter most to customers, on Metis today and on Europa, and turn each lesson into a platform improvement.

Skills

Strong C++

Python

Computer architecture

Operating systems

Parallel programming

Heterogeneous computing

Tools

PyTorch

ONNX

MLIR

FX

Job description

Axelera AI is seeking a Staff AI Application Engineer to build the model enablement platform and decide how each model runs across the Metis or Europa AIPU, the host CPU and accelerators. The role covers the frontend, graph compiler, runtime and application pipeline, with model bring-up spanning frontend and graph sides.

You will work with AI coding agents in daily tasks and collaborate with field engineers, tech leads and managers who want to stay hands-on.