AIML - Apple Foundation Models: ML Research Engineer, RL & Agentic Reasoning
Core
Develop and scale reinforcement learning methods to enhance Apple's foundation models for agentic tool use, planning, and reasoning, directly impacting features like Siri.
Role type
Senior IC machine learning research engineer (RL & agentic reasoning)
Builds
Foundation models with improved reasoning, instruction following, and multi-turn dialogue capabilities
Domain
Generative AI / Large Language Models / Reinforcement Learning
Deliverable
production ML models
Required skills
Reinforcement learning, Deep learning, Large-scale model training, Python, Distributed training, Reward modeling, Simulation environment design
Preferred skills
Agentic tool use, Planning algorithms, Retrieval integration, Data curation, Evaluation frameworks, Safety guardrails
Technologies
PyTorch, JAX
Responsibilities
Develop RL methods to reduce hallucinations and improve reasoning; Design and train agents with tool use and API integration; Build reward models, evaluators, and simulation environments; Run large-scale experiments and translate findings into research contributions and product improvements
Seniority
Senior, hands-on IC