Senior Research Engineer, Training Data Infrastructure in Foundation Models

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$184,700–$324,800 / yr
Posted
2 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $211k
This role $255k
$155k most similar roles pay here $343k

This role pays more than 87% of similar roles. Most pay $175,750–$246,987 — the shaded band above. At the midpoint, this role pays about $255k versus about $211k 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 2305 open roles on FindRole.

Listed pay typically runs $175,000–$280,000 across 1873 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Senior Research Engineer, Training Data Infrastructure in Foundation Models

As a Senior Research Engineer, Training Data Infrastructure in Foundation Models, you will join a team focused on the full training lifecycle of frontier foundation models. You will work at the intersection of software engineering and machine learning research to transform theoretical observations into scalable systems for data acquisition, processing, and repository management. Your daily responsibilities include architecting high-throughput distributed ingestion systems for petabytes of multimodal data, optimizing data formats for efficient access, and building automated pipelines for filtering, deduplication, and decontamination. You will also develop tools for scaling experiments and data mixture optimization. The role requires expert proficiency in Python and C++, along with experience in distributed systems, performance engineering, and cloud infrastructure like GCP or AWS. You will solve complex problems regarding reward modeling, training efficiency, and high-quality data curation for large model training.

What you'll do

  • Design and implement high-throughput distributed systems to ingest petabytes of multimodal data from diverse sources.
  • Manage the lifecycle of large-scale datasets across storage and high-performance file systems while optimizing data formats for training.
  • Build and maintain distributed data processing workflows on cloud infrastructure using advanced frameworks.
  • Implement sophisticated filtering, semantic deduplication, and synthetic data generation to ensure high-quality training content.
  • Develop automated systems to detect and remove benchmark leakage from training corpora.
  • Engineer robust data governance and privacy solutions to ensure compliance with regulatory standards.
  • Build tools to support systematic data mixture optimization and empirical scaling experiments for research teams.

What we're looking for

  • Bachelor's degree in Computer Science, Electrical Engineering, or Mathematics.
  • 4+ years of software engineering experience focusing on Data Infrastructure, Distributed Systems, or AI/ML Engineering.
  • Expert fluency in Python and strong competence in system languages such as C++.
  • Extensive experience architecting scalable data systems on major public cloud platforms like GCP using tools like Apache Beam.
  • Deep experience profiling and optimizing high-throughput data systems and debugging distributed bottlenecks.
  • Experience working closely with ML research organizations to translate research results into engineering implementations (preferred).
  • Familiarity with the lifecycle of modern LLM training, end-to-end workflows, and underlying system architecture (preferred).
  • Experience processing complex data modalities including source code, images, videos, and audio (preferred).

More like this

Similar roles

Machine Learning Researcher, Foundation Models

Apple Inc

Cupertino, CA 2 days ago $150,400$277,600
Large Language Models Deep Learning Python PyTorch JAX TensorFlow Reinforcement Learning Pallas Triton Kernel Optimization Data Research LLM) architecture Mid-training On-policy Distillation

AIML Researcher/Engineer, Foundation Model Post-Training

Apple Inc

Cupertino, CA 114 days ago $184,700$324,800
LLMs Reinforcement Learning Python PyTorch JAX Distributed Training Deep Learning Data Filtering Curriculum Learning Synthetic Data Generation Model Steering Preference Optimization Evaluation Methodologies

AIML Researcher/Engineer, Foundation Model Post-Training

Apple Inc

New York, NY 114 days ago $184,700$324,800
LLM Reinforcement Learning Python PyTorch JAX Distributed Training Data Filtering Curriculum Learning Synthetic Data Generation Model Steering Preference Optimization Deep Learning Machine Learning

AIML Researcher/Engineer, Foundation Model Post-Training

Apple Inc

Seattle, WA 107 days ago $175,000$308,500
LLM Reinforcement Learning Python PyTorch JAX Distributed Training Deep Learning Data Filtering Synthetic Data Generation Curriculum Learning Preference Optimization Machine Learning