Application Engineer

• 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

Explore Staff AI Application Engineer – Heterogeneous Computing at Axelera AI. Review requirements and application details.