Remote AI Application Engineer: Heterogeneous Architect

Axelera AI; Portugal; Remote; EUR 110,000 - 150,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 to build and optimize our model enablement platform across Metis/Europa AIPU, CPU and accelerators. You will shape how models are brought up, scheduled and run, balancing frontend, graph compiler, runtime and application pipelines.

You will work with AI coding agents daily and collaborate with engineers across the graph compiler, runtime and pipeline teams to achieve peak end-to-end performance on diverse hardware.

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.
• Model ingestion: Capture, analyse and split PyTorch/ONNX models across the AIPU components and accelerators.
• Heterogeneous execution: Extend the split to large models, cascades and multi-model pipelines.
• End-to-end performance: Work with pipeline and runtime teams to optimize host-device overlap, zero-copy, batching and scheduling.
• Critical models: Study models that matter to customers and drive platform improvements.

Skills

Systems engineering

Heterogeneous computing

C++

Python

Graph representations

AI coding agents

Tools

MLIR

TorchInductor

ONNX

CUDA

Job description

Axelera AI is seeking a Staff AI Application Engineer to build and optimize our model enablement platform across Metis/Europa AIPU, CPU and accelerators. You will shape how models are brought up, scheduled and run, balancing frontend, graph compiler, runtime and application pipelines.

You will work with AI coding agents daily and collaborate with engineers across the graph compiler, runtime and pipeline teams to achieve peak end-to-end performance on diverse hardware.