Machine Learning Researcher: Agentic Post-Training for On-Device Models - Jobs - Careers at Apple
- Advance agentic post-training for on-device models. Develop supervised fine-tuning, reinforcement learning, preference optimization, and distillation methods that improve reasoning and multi-step tool use in on-device LLMs.
- Design training data and interactive environments. Build scalable pipelines for trajectory generation, curriculum construction, and hard-example mining. Develop environments that support learning through tool use and multimodal interaction.
- Design reward functions, verifiers, and graders for complex agent tasks, addressing long-horizon credit assignment, supervision quality, and gaps between reward scores and actual task success.
- Improve capability under deployment constraints. Research methods that balance task success with latency, memory, energy, and tool-use costs, including structured or compressed reasoning and adaptive computation.
- Build rigorous evaluations and diagnostics. Measure task success, generalization, reliability, and efficiency. Analyze model outputs and agent trajectories to distinguish modeling limitations from data, reward, environment, and infrastructure issues, and turn findings into targeted improvements.
- Translate research into Apple experiences. Collaborate with software and hardware teams to integrate and validate agent capabilities under realistic on-device constraints.
- Drive external research impact. Contribute to research projects from initial hypothesis through experimental validation, publish high-impact technical reports as a core author, and release open-source models that advance agentic post-training and support the broader research community. Example technical reports: “Internalized Visual Thinking for Video Reasoning, Structured Reasoning for Efficient Agentic Search.”
- MS in Computer Science, Machine Learning, Artificial Intelligence, Computer Vision, Natural Language Processing, or a related field, or equivalent practical experience.
- A strong research track record demonstrated by either first-author publications at top-tier conferences (such as NeurIPS, ICML, ICLR, CVPR, ICCV, ECCV, ACL, or EMNLP) or substantial hands-on experience training frontier models, with contributions relevant to foundation models, reinforcement learning, multimodal learning, or agentic systems.
- Strong programming skills in Python and proficiency with deep learning frameworks such as PyTorch.
- Demonstrated ability to formulate research hypotheses, implement training and evaluation pipelines, conduct controlled experiments, and draw well-supported conclusions.
- PhD in Computer Science, Machine Learning, Artificial Intelligence, Computer Vision, Natural Language Processing, or a related field, or equivalent practical experience. Research focus on foundation models, reinforcement learning, agentic systems, multimodal reasoning, or efficient machine learning.
- Significant contributions to high-impact technical reports or open-source model releases in agentic post-training, with substantial ownership of research methodology, model training, experimental validation, and technical communication.
- Experience developing or training agents for tool use and long-horizon tasks.
- Experience designing interactive training environments, trajectory-generation pipelines, reward models, or scalable agent evaluations.
- Experience improving smaller models through compute-aware post-training, with an understanding of on-device latency, memory, and energy constraints.
- Ability to independently drive research projects and collaborate effectively across research and engineering teams.
Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant
At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.
Learn about accessibility in Apple’s workplace
Learn about reasonable accommodations for job applicants
Apple accepts applications to this posting on an ongoing basis.
Required Skills
Required Languages
🇬🇧 English