AIML - ML Research Engineer, MIND
Job Description
Summary
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.
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
- 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
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 on CUDA/GPU and programming
