AIML - ML Research Engineer, MIND

at  Apple

SUMMARY

Posted: Oct 27, 2024
Weekly Hours: 40
Role Number:200575889
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 hand-on experience in software development to join our team. Our mission is to advance the state-of-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

• Research and develop ML optimization methods for efficient on-device ML - Materialize the ideas/concepts and demonstrate the practical value in an industrial environment - Work with HW/SW teams to productize the outcome of the work

• 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 on CUDA/GPU and programming

Please refer the Job description for details