AIML - Distinguished Engineer, Foundation Models

Apple•Cupertino, CA

About The Position

This team builds frontier foundation models that power intelligent experiences at Apple, working across the full training lifecycle, including pre-training and developing mid-training approaches. The models are engineered specifically for Apple silicon and optimized for private, personal, and deeply integrated OS experiences. The role involves solving frontier problems in reward modeling to resist reward hacking, handling sparse and delayed rewards in agentic settings, and aligning models reliably across open-ended creative tasks and precise, action-taking workflows. The team believes that the most interesting problems and important breakthroughs in deep learning research arise from applying learning to real-world use cases. You will collaborate with a close-knit and fast-growing team of world-class engineers and scientists to tackle challenging problems in foundation models and deep learning, including natural language processing, multi-modal understanding, and combining learning with knowledge.

Requirements

Demonstrated expertise in deep learning with a publication record in relevant conferences (e.g., NeurIPS, ICML, ICLR, CVPR, ICCV, ECCV, KDD, ACL, ICASSP, InterSpeech) or a track record in applying deep learning techniques to products
Proficient programming skills in Python and one of the deep learning toolkits such as JAX, PyTorch, or Tensorflow
Ability to work in a collaborative environment

Nice To Haves

Web-scale information retrieval
Human-like conversation agent
Multi-modal perception for existing products and future hardware platforms
On-device intelligence and learning with strong privacy protections
PhD, or equivalent practical experience, in Computer Science, or related technical field