AIML Researcher/Engineer - Foundation Model Post-Training
Core
Design and iterate end-to-end post-training strategies (including Reinforcement Learning) to transform large language models into intelligent assistants for Apple products.
Role type
Senior IC machine-learning researcher/engineer (foundation model post-training)
Builds
Large-scale frontier foundation models powering billions of Apple devices
Domain
Deep learning, Large Language Models (LLMs), Reinforcement Learning
Deliverable
production ML models
Required skills
Deep learning expertise, LLM post-training, Reinforcement Learning, Python programming, JAX or PyTorch, distributed training, transformers architectures, automated data filtering, curriculum learning, evaluation methodology design
Preferred skills
Training state-of-the-art large models at scale, improving model performance on complex reasoning tasks (math, coding, logic), cross-functional collaboration
Technologies
Python, JAX, PyTorch, Transformers
Responsibilities
Design and iterate on end-to-end post-training strategies; Pioneer novel algorithms for preference optimization, model steering, and safety; Drive data strategy for high-quality human and synthetic data generation; Design robust evaluation methodologies; Partner with pre-training and product teams to translate requirements into model capabilities
Seniority
Senior, hands-on IC
