On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning
Core
Building an end-to-end developer experience and core runtime for machine learning execution across Apple devices using an MLIR-based compiler stack.
Role type
Senior IC ML compiler engineer (on-device)
Builds
MLIR-based compiler stack, runtimes, and optimization toolkits for heterogeneous compute
Domain
Consumer electronics / Machine Learning / Compiler Infrastructure
Deliverable
production ML models
Required skills
MLIR-based compilers, C++, ML model architectures, PyTorch
Preferred skills
Swift, GPU/CPU/Neural Engine programming paradigms, kernel writing for ML execution
Technologies
MLIR, PyTorch, C++, Swift
Responsibilities
Inspire changes in the MLIR-based compiler to target improved runtime performance; Propose upstream changes in MLIR to support new hardware features; Own core pieces of the compiler stack enabling heterogeneous compute; Work with hardware, software, and performance teams to optimize execution across Apple devices.
Seniority
Mid-Senior, hands-on IC
