Staff AI Application Engineer – Heterogeneous Computing

Job not on LinkedIn

🔥 0 minutes ago

🇧🇪 Belgium – Remote

⏰ Full Time

🔴 Lead

💻 Application Engineer

👻 Ghost score 13%

What it means: how likely this posting is sitting open without active hiring.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Axelera AI

WebsiteLinkedInAll Job Openings

51 - 200 employees

Founded 2021

🏭 Manufacturing

💼 Consulting

📦 Logistics

Manufacturing • Consulting • Logistics

Axelera AI is a leading provider of AI hardware acceleration technology tailored for generative AI and computer vision inference. The company offers advanced AI processing units designed to integrate seamlessly into innovations, focused on delivering high-performance, cost-effective, and energy-efficient AI solutions. Axelera AI's flagship product, the Metis AI Processing Unit, accelerates inference at the edge, making it ideal for applications like video analysis and multi-channel video analytics. The company is recognized for its deep tech innovations and is actively contributing to the advancement of AI by providing state-of-the-art AI acceleration capabilities for edge computing.

📋 Description

• Build the model enablement platform and agent harness used by engineers and AI agents to bring up models • Develop reusable tools to partition, compile, run and profile models • Provide real-hardware environments and closed-loop runs that measure changes against references • Integrate differential testing and regression detection into CI • Own how PyTorch or ONNX models are captured, analysed and split across AIPU units, host CPU and host accelerators • Build placement tags that users can read and override • Extend heterogeneous execution to large models, cascades and multi-model pipelines • Optimize end-to-end performance through host/device overlap, zero-copy buffers, batching, multi-core scheduling and multi-input models • Profile device time, host time and data movement and validate speedups in the full pipeline • Study critical customer models on Metis and Europa and convert findings into platform improvements • Collaborate with graph compiler, application and runtime teams

🎯 Requirements

• 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 • Ability to reason about compute, memory bandwidth, latency and data movement • Solid fundamentals in computer architecture, operating systems and parallel programming • Familiarity with ML graph representations and compiler concepts, including graph capture, lowering, operator fusion, partitioning and memory planning • Strong C++ and Python • Daily use of AI coding agents on real engineering work, with judgment to review and test their output • Hands-on experience with MLIR or another compiler framework preferred • Experience with an inference runtime or execution engine preferred • Experience with GPU compute or SIMD optimization on Arm or x86 preferred • Experience with large-model execution techniques preferred • Experience building agent environments, evaluation harnesses or verification frameworks preferred

🏖️ Benefits

• Pension plan • Extensive employee insurances • Option to get company shares • Flexible working arrangement • Open culture supporting creativity and continual innovation • Collaborative ownership and freedom with responsibility • Equal opportunity and inclusive environment