On-Device ML Infrastructure Engineer (CoreML Runtime)

Imagine being at the forefront of an evolution where powerful AI meets the elegance of Apple silicon. The On-Device Machine Learning team transforms groundbreaking research into practical applications, enabling billions of Apple devices to run powerful AI models locally, privately, and efficiently.

• Architect and maintain the on-device graph compiler, runtime, and kernels for delivering ML operators
• Develop production-critical system software for implementing ML models on Apple Silicon
• Proactively identify and resolve functionality gaps
• Optimize model execution for various system objectives like performance, energy efficiency, and thermal management

• Masters or equivalent experience in Computer Sciences, Engineering, or related subject area
• Highly proficient in C++ or Swift
• Familiarity with Python
• Experience with any compiler stack (MLIR/LLVM/TVM/...)
• Familiarity with Operating Systems, embedding programming, parallel programming
• Sound understanding of ML fundamentals, including common architectures such as Transformers

• Comprehensive medical and dental coverage
• Retirement benefits
• Range of discounted products and free services
• Reimbursement for certain educational expenses
• Discretionary bonuses or commission payments
• Relocation

Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities