Machine Learning Research Lead, RL Agents

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$216,200–$394,000 / yr
Posted
6 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $225k
This role $305k
$160k most similar roles pay here $419k

This role pays more than 93% of similar roles. Most pay $194,375–$254,750 — the shaded band above. At the midpoint, this role pays about $305k versus about $225k for comparable roles.

Based on 240 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 1957 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1579 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Machine Learning Research Lead, RL Agents

AIML - Machine Learning Research Lead, RL Agents, MLR serves as a hands-on research lead within the MLR group, managing a small team of senior researchers and engineers. The role focuses on driving reinforcement learning and post-training for agentic AI, specifically targeting tool calling, coding, and multi-step interactive tasks. Responsibilities include building synthetic data and task-generation pipelines, developing infrastructure for core RL research, and designing methods for on-device and hybrid execution. The candidate will work with technologies including generative foundation models across multiple modalities, self-supervised learning, and evolutionary methods. Key technical areas involve reward and preference optimization, verifier design, and model scaling. This role addresses the challenge of creating capable agents that function within specific hardware constraints while advancing first-principles research in vision-language modeling and world models for planning and policy.

What you'll do

  • Lead research on reinforcement learning and post-training for agentic capabilities including tool calling, coding, and multi-step tasks.
  • Build and own synthetic data pipelines and interactive environments to create training curricula for capable agents.
  • Develop and manage the codebases and infrastructure required for core RL research efforts.
  • Manage and mentor a small team of senior researchers while shaping their shared research direction.
  • Perform hands-on technical work including running experiments, writing code, and solving complex engineering problems.
  • Optimize post-training methods to ensure compatibility with on-device and hybrid compute constraints.
  • Publish original research in top venues and engage with the broader academic community.

What we're looking for

  • PhD in machine learning or a related field, or equivalent research experience.
  • 7-10+ years of research experience beyond a PhD in industry or as an academic research lead.
  • Strong track record in RL and/or post-training of large models through publications, open-source contributions, or shipped systems.
  • Leadership experience setting research direction over multiple years and directing the work of others.
  • Experience owning ML infrastructure, frameworks, and codebases including open-source research frameworks or environment suites.
  • Strong engineering skills with comfort working hands-on in large training codebases.
  • Experience taking research from idea to product or production impact (preferred).
  • Familiarity with efficiency-aware modeling, such as small models, MoE, quantization, or hardware-aware design (preferred).

More like this

Similar roles

Machine Learning Researcher

Apple Inc

Cupertino, CA 49 days ago $150,400$277,600
Machine Learning Python PyTorch Deep Learning Linear Algebra Statistics Multi-modal Multi-task Reasoning open-source

Machine Learning Researcher

Apple Inc

Seattle, WA 49 days ago $142,300$263,300
Machine Learning Python PyTorch Deep Learning Linear Algebra Statistics Multi-modal Multi-task Reasoning open-source

Senior Machine Learning Research Engineer, AIML

Apple Inc

Cupertino, CA 153 days ago $184,700$324,800
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI On-device Inference ML Infrastructure software–hardware co-design Parallelization Algorithms Foundation Models Machine Learning Computational Science
3+ yrs exp

Senior Machine Learning Research Engineer, AIML

Apple Inc

Cupertino, CA 64 days ago $184,700$324,800
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI On-device Inference ML Infrastructure software–hardware co-design Parallelization Algorithms Foundation Models Machine Learning Research
3+ yrs exp