AIML - ML Research Engineer, MIND

The Machine Intelligence, Neural Design (MIND) team at Apple is leading company-wide innovation on HW/SW co-design for efficient inference. They are seeking a highly motivated Research Engineer with a strong background in ML systems and hands-on experience in software development. The ideal candidate will push existing boundaries with out-of-the-box thinking and materialize ideas to impact both Apple partners and broader research communities.

Key responsibilities include:

• Researching and developing ML optimization methods for efficient on-device ML
• Materializing ideas/concepts and demonstrating practical value in an industrial environment
• Working with HW/SW teams to productize the outcome of the work

Minimum Qualifications:

• Expert in Python/PyTorch programming
• Quality publications in ML optimization (e.g., NeurIPS, ICML, ICLR, CVPR, ACL, etc.)
• Excellent communication and writing skills
• MS, PhD, or equivalent experience in Computer Science/Computer Engineering or related fields

Preferred Qualifications:

• Experience with Jax programming
• Experience with Large-scale model training/tuning
• Experience with backend compilation, HW/SW co-design, and/or performance optimization
• Understanding ML accelerations and architectures
• Understanding of CUDA/GPU and programming

Apple offers a comprehensive compensation package, including base pay, potential stock awards, and various benefits such as medical and dental coverage, retirement benefits, and educational reimbursement opportunities.