Axelera

Senior/Staff AI Application Engineer - Heterogeneous Computing

Hybrid/Remote - Europe (incl. UK). Posted 19 days ago

The role

You will build a model enablement platform and decide how models run across the Metis or Europa AIPU, the host CPU, and host accelerators. The role involves architecture and hands-on work across the model frontend, graph compiler, runtime, and application pipeline.

What you will do

  • Build the model enablement platform and agent harness
  • Create reusable tools to partition, compile, run and profile
  • Own how a PyTorch or ONNX model is captured, analysed and split across units
  • Extend heterogeneous execution to large models, cascades and multi-model pipelines
  • Bring each model to its best end-to-end performance
  • Profile device time, host time and data movement
  • Study critical models to turn lessons into platform improvements

Required qualifications

  • 8+ years of engineering experience in systems, compilers, runtimes or performance, including system-level design ownership
  • 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

Nice to have

  • Hands-on experience with MLIR or another compiler framework (TVM, XLA, TorchInductor, IREE)
  • Experience with an inference runtime or execution engine (ONNX Runtime execution providers, a custom NPU or DSP runtime)
  • Experience with GPU compute (CUDA, OpenCL, Vulkan) or SIMD optimization on Arm or x86
  • Experience with large-model execution techniques such as offloading, expert caching, speculative decoding, pipelining across devices
  • Experience building agent environments, evaluation harnesses or verification frameworks (differential testing, regression detection), or MCP-based tooling

Benefits

  • Pension plan
  • Extensive employee insurances
  • Option to get company shares

About the company

Axelera AI is a deep-tech company creating a next-generation AI platform. It has a team of 250+ employees and is headquartered at the High Tech Campus in Eindhoven, Netherlands.