AIML - ML Research Engineer, MIND

The Machine Intelligence, Neural Design (MIND) team is leading Apple-wide innovation on HW/SW co-design for efficient inference. We are looking for a highly motivated Research Engineer with a strong background in ML systems and hands-on experience in software development to join our team. Our mission is to advance the state-of-the-art in ML optimization techniques used in efficient inference. Thus, the ideal candidate for this position is a researcher and engineer who will relentlessly push the existing boundaries with out-of-the-box thinking and materialize their ideas with the very detail required to impact both Apple partners and broader research communities.

Description

1. Research and develop ML optimization methods for efficient on-device ML.
2. Materialize the ideas/concepts and demonstrate the practical value in an industrial environment.
3. Work 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 programming.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure 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. Learn more about your EEO rights as an applicant.