Job Description
Summary
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
- 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