Staff AI Application Engineer - Heterogeneous Computing (Fully Remote)

Axelera AI; United States; Remote; USD 180,000 - 230,000; Full time

Benefits offered by this job

Pension plan

Insurance

Company shares

Open culture promoting diversity

Job summary

Axelera AI builds the next-generation AI platform, delivering 3-5x efficiency gains with a strong, open culture across 21 countries. We are seeking a hands-on engineer to own the model enablement platform and the agent harness for Voyager SDK, while driving architecture across frontend, compiler, runtime, and pipeline to enable cutting-edge AI workloads.

You will build tools to partition, compile, run, and profile models with verification, enable heterogeneous execution on AIPU/CPU/accelerators

Responsibilities

• Build tools for partitioning, compiling, running, and profiling models with verification.
• Enable heterogeneous execution on AIPU, CPU, and accelerators with offloading.
• Optimize performance by profiling host/device and validating full pipeline.

Job description

• Own the model enablement platform and agent harness for Voyager SDK.
• Drive architecture across frontend, compiler, runtime, and pipeline.

Key Responsibilities:

• Build tools for partitioning, compiling, running, and profiling models with verification.
• Enable heterogeneous execution on AIPU, CPU, and accelerators with offloading.
• Optimize performance by profiling host/device and validating full pipeline.

What We Offer:

• Attractive compensation with pension plan, insurance, and company shares.
• Open culture supporting creativity, innovation, and diversity.

Axelera AI builds the next-generation AI platform, with Metis delivering 3-5x efficiency gains to support human advancement. With $450M raised and 250+ employees (60+ PhDs), it offers an open, innovative, inclusive culture spanning 21 countries.