Foundation Models ML Engineer — Data, Diffusion & Datasets

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

Benefits offered by this job

RSU awards

Education reimbursement

Relocation assistance

Health benefits

Job summary

Apple, Inc. in Seattle is seeking a Machine Learning Engineer focused on data for foundation models. You will build high-quality datasets for training and apply data ablation to improve diffusion and autoregressive models, aligning with Apple’s product goals.

You will design data annotation, enrichment, curation and synthetic data pipelines using traditional and generative AI tools to enhance model quality across specific dimensions and platforms.

Qualifications

• MS or PhD in Computer Science or related field.
• 3+ years of experience improving modern image generation or image understanding models through data ablation experiments.
• Proficiency with ML frameworks such as PyTorch or JAX and large-scale data frameworks like Hadoop or Ray.
• Experience in academic or industrial research environment.
• Excellent communication, bias-for-action and ability to drive clarity in ambiguous circumstances.

Skills

ML fundamentals

Data ablation

Scientific communication

Ambiguity handling

Education

MS or PhD in Computer Science or related field

Tools

PyTorch

JAX

Hadoop

Ray

Job description

Apple, Inc. in Seattle is seeking a Machine Learning Engineer focused on data for foundation models. You will build high-quality datasets for training and apply data ablation to improve diffusion and autoregressive models, aligning with Apple’s product goals.

You will design data annotation, enrichment, curation and synthetic data pipelines using traditional and generative AI tools to enhance model quality across specific dimensions and platforms.