On-Device ML Compiler Engineer
Core
Develop and improve MLIR-based compiler infrastructure to optimize machine learning model compilation and execution across Apple's Neural Engine, GPU, and CPU.
Role type
Senior IC ML compiler engineer
Builds
Optimized ML model execution on Apple devices
Domain
Mobile computing / Machine Learning infrastructure
Deliverable
production ML models
Required skills
MLIR-based compiler development, C++, machine learning model architectures, PyTorch or related training frameworks
Preferred skills
Swift, GPU/CPU/Neural Engine programming paradigms, writing kernels for ML model execution
Technologies
MLIR, PyTorch, C++, Swift
Responsibilities
Develop and improve MLIR-based compiler infrastructure, write kernels and compiler optimizations, support model optimization, transformation, execution, debugging, profiling, and analysis, collaborate with model authoring, runtime, and performance teams
Seniority
Mid-level, hands-on IC
