Staff AI Application Engineer - Heterogeneous Computing
Posted Updated
Description
You will build model enablement tools and agent workflows for model bring-up. You will own model ingestion and partitioning across accelerators, extend heterogeneous execution for large and multi-model workloads, optimize end-to-end performance, and turn lessons from critical models into platform improvements.
Responsibilities
- Build model enablement platforms, agent harnesses and verification workflows
- Own model capture, analysis and partitioning across AIPUs, CPUs and host accelerators
- Extend heterogeneous execution for large models and multi-model pipelines
- Optimize end-to-end performance across host and device work
- Profile device time, host time and data movement
- Study critical customer models and turn findings into platform improvements
Requirements
- 8+ years of engineering experience in systems, compilers, runtimes or performance
- System-level design ownership
- Heterogeneous computing
- Computer architecture
- Operating systems
- Parallel programming
- ML graph representations
- Compiler concepts
- C++
- Python
- AI coding agent usage
Benefits
- Pension plan
- Employee insurances
- Company share option