AI Application Engineer: Heterogeneous Computing & Offload

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

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