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 restricted stock unit awards
• Employee Stock Purchase Plan
• Discretionary bonuses or commission payments
• Relocation assistance

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.