ML Researcher, Apple Foundation Models

Apple Inc

Actively hiring Verified listing
New York City Metro Area, NY Posted 9 days ago $181,100–$318,400 / year
View original post Log in to save

At a glance

AI generated

TL;DR

As an ML Researcher at Apple Foundation Models in New York City, you will join a team dedicated to developing advanced foundation models tailored for Apple silicon, focusing on privacy, personalization, and seamless integration into the OS. Your primary responsibilities include training models via reinforcement learning (RL) to enhance reasoning, coding, and agentic capabilities, addressing challenges like verifiable rewards for mathematical reasoning and multi-turn RL for coding agents. You will work with cutting-edge technologies such as JAX, PyTorch, or TensorFlow, applying your expertise in deep learning and reinforcement learning to solve complex problems that directly impact millions of users. This role requires a PhD in Computer Science or related field, along with experience in top ML/NLP conferences or practical application of deep learning techniques to products.

Skills

Python PyTorch Tensorflow JAX Reinforcement_learning RLHF GRPO PPO RLVR reward_modeling SWE-Bench sparse_attention context_compression on_policy_distillation reward_tilted_distillation cross_stage_distillation

What you'll do

  • Develop frontier foundation models for Apple silicon, optimizing for privacy and personalization.
  • Train models using reinforcement learning to enhance reasoning and coding capabilities.
  • Work on reward modeling to prevent reward hacking in agentic settings.
  • Conduct research on scaling laws for RL compute allocation and progressive alignment.
  • Build autonomous coding agents that operate within real repositories and handle multi-step workflows.
  • Engage in applied ML research with direct product impact, shipping innovations to millions of users.

What we're looking for

  • Demonstrated expertise in deep learning with publications at top ML or NLP conferences.
  • Proficient programming skills in Python and experience with JAX, PyTorch, or TensorFlow.
  • PhD in Computer Science or related technical field, or equivalent practical experience.
  • Experience with reinforcement learning for large language models (RLHF, GRPO, PPO).
  • Skills in code generation, coding agents, and agentic systems development.

Market check

Salary context

This $181,100–$318,400 range sits above 88% of similar postings on FindRole.

Peer median band

$129,250–$225,000

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$142,400–$224,325

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

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 43 open roles on FindRole.

Listed pay typically runs $147,400–$272,100 across 35 roles with salary data.

Most-posted roles

View all roles at Apple Inc

More like this

Similar roles

ML Researcher, Apple Foundation Models

Apple Inc

New York City Metro Area, New York, US 9 days ago $181,100–$318,400
Python PyTorch Tensorflow JAX Reinforcement_learning RLHF GRPO PPO RLVR reward_modeling SWE-Bench sparse_attention context_compression on_policy_distillation reward_tilted_distillation cross_stage_distillation

ML Researcher, Apple Foundation Models

Apple Inc

San Francisco Bay Area, California, US 9 days ago $181,100–$318,400
Python JAX PyTorch Tensorflow Reinforcement_learning RLHF GRPO PPO RLVR reward_modeling RL_scaling_laws code_generation agentic_coding multi_turn_RL tool_use_planning long_horizon_task_execution user_simulation on_policy_distillation reward_tilted_distillation cross_stage_distillation sparse_attention context_compression

Researcher/ML Engineer

Adobe

San Jose, US 10 days ago $183,300–$265,350
Python C/C++ PyTorch CoreML LiteRT ONNX iOS Android GPU CI/CD

Machine Learning Researcher

Qualcomm

San Diego, Ca,Us, US 164 days ago $159,100–$238,700
Python TensorFlow PyTorch Keras Hugging Face Transformers CUDA C++ Git Jupyter Notebook CI/CD Docker GitHub NeurIPS ICML ICLR CVPR ICCV ACL EMNLP

Senior AI/ML Scientist

General Motors (GM)

Remote (Milford Proving Ground - Bldg 16 - Hybrid, Engrg, Brake Systems, US) 38 days ago
Python PyTorch TensorFlow HuggingFace Transformers Scikit-learn XGBoost Docker Kubernetes CI/CD MLOps FastAPI Ray Prometheus Grafana PostgreSQL AWS Simulink AMESIM
Remote