Senior AI App Engineer, Heterogeneous Compute (Remote)

Axelera AI; España; Remote; EUR 110,000 - 160,000; Full time

Benefits offered by this job

Pension plan

Extensive employee insurances

Company shares

Job summary

Axelera AI is seeking a Staff AI Application Engineer in the Model Enablement team. You will build the model enablement platform and decide how models run across Metis/Europa AIPU, host CPUs and accelerators.

The role combines architecture work with hands-on implementation across frontend, graph compiler, runtime and pipeline. The team uses AI coding agents daily, and the ideal candidate brings strong C++/Python skills, extensive experience with heterogeneous compute, and a track record of

Qualifications

• 8+ years of engineering experience in systems, compilers, runtimes or performance with ownership of system-level design.
• Hands-on experience with heterogeneous computing across CPU, GPU, DSP, NPU or FPGA including data movement considerations.
• Strong 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.
• Proficiency in C++ and Python; comfortable working with AI coding agents.

Responsibilities

• Build and maintain the model enablement platform and agent harness for model bring-up and verification.
• Capture, analyze and split PyTorch/ONNX models across AIPU, host CPU and accelerators with clear placement tags.
• Extend heterogeneous execution to large models and multi-model pipelines with caching and offloading.
• Collaborate with graph compiler and runtime teams to maximize end-to-end performance (overlaps, zero-copy, batching, scheduling).
• Study high-impact models and drive platform improvements through experimental validation.

Skills

8+ years systems engineering

Heterogeneous compute

Computer architecture

C++

Python

AI agents usage

ML graph representations

Graph compiler concepts

Tools

MLIR

ONNX

CUDA

Vulkan

OpenCL

Job description

Axelera AI is seeking a Staff AI Application Engineer in the Model Enablement team. You will build the model enablement platform and decide how models run across Metis/Europa AIPU, host CPUs and accelerators.

The role combines architecture work with hands-on implementation across frontend, graph compiler, runtime and pipeline. The team uses AI coding agents daily, and the ideal candidate brings strong C++/Python skills, extensive experience with heterogeneous compute, and a track record of