ML Engineer: Foundation Models Data Pipelines

Apple Inc.; Seattle (WA); On-site; USD 150,000 - 278,000; Full time

Benefits offered by this job

Comprehensive medical and dental

Retirement benefits

Apple stock programs

Tuition reimbursement

Job summary

Apple Inc. in Cupertino, CA, seeks a Machine Learning Engineer to build high-quality datasets for foundation model training.

You will design data annotation, enrichment, curation, and synthetic data generation pipelines to improve diffusion and autoregressive models across Apple products. Ideal candidates have MS/PhD, 3+ years of ML data experience, and proficiency with PyTorch or JAX, plus experience with Hadoop or Ray.

Qualifications

• MS/PhD in CS or related field required.
• 3+ years improving image generation/understanding via data ablation.
• Experience with PyTorch or JAX and large-scale data frameworks.

Responsibilities

• Design data annotation, enrichment, and curation pipelines for foundation model training.
• Build synthetic data generation workflows to improve model quality.
• Apply ML research to improve diffusion or autoregressive models with large-scale data.

Skills

Data ablation experiments

PyTorch

JAX

Hadoop

Ray

Communication skills

Research experience

Education

M.S. or PhD in Computer Science or related field

Tools

GenAI tooling

Distributed training

Job description

Apple Inc. in Cupertino, CA, seeks a Machine Learning Engineer to build high-quality datasets for foundation model training.

You will design data annotation, enrichment, curation, and synthetic data generation pipelines to improve diffusion and autoregressive models across Apple products. Ideal candidates have MS/PhD, 3+ years of ML data experience, and proficiency with PyTorch or JAX, plus experience with Hadoop or Ray.