Senior ML Platform Engineer — GenAI & Data at Scale

Apple Inc.; Seattle (WA); On-site; USD 175,000 - 308,500; Full time

Job summary

Apple Inc. in Seattle, WA, is seeking engineers and researchers to join the Cloud AI Platform team, delivering data systems and compute for next‑gen ML workloads and GenAI applications across Apple products.

You’ll help scale data pipelines, embeddings, and model workflows, from experimentation to production. Ideal candidates have strong ML foundations, experience with large distributed systems, and proficiency in Python, Java or Go, plus a track record in real‑world ML infrastructure and

Qualifications

• Strong foundation in machine learning from end-to-end workflow to deployment.
• Experience building and running large-scale distributed systems.
• Familiarity with modern generative techniques (transformers, diffusion, RAG).
• Proven production experience with data and ML infrastructure.
• Experience with fine-tuning, model optimization, scalable inference.
• Exposure to GenAI applications and fast-moving tech stacks.
• Ability to deploy, troubleshoot, and scale production environments.
• Designing scalable, highly available, easy-to-use systems.

Responsibilities

• Design and build the platform behind Apple’s largest model builds — ingestion, immutable versioning, lineage, and governance across data at petabyte scale.
• Develop and evolve Python SDKs and core data libraries for ML engineers to access, transform, and load datasets.
• Build high-throughput data access and loading primitives for large GPU fleets.
• Operate distributed data pipelines across Spark, Daft, and Rust-based systems for ingestion and preparation.
• Optimize components for tight integration with PyTorch, JAX, and TensorFlow in the model dev loop.
• Onboard new data sources with researchers and product teams for GenAI workloads.
• Ensure governance, privacy controls, and end-to-end data lineage across datasets.
• Drive efficiency, reliability, and automation across data and control planes.

Skills

Large-scale distributed systems

Python

Java

Go

Machine learning fundamentals

Data pipelines

SQL

Collaboration & communication

End-to-end ML workflow

Ambiguity navigation

Education

BS, MS, or PhD in Computer Science / Engineering

Equivalent practical experience

Tools

Docker

Kubernetes

Spark

Daft

Polars

DuckDB

TensorFlow

PyTorch

JAX

Rust

Job description

Apple Inc. in Seattle, WA, is seeking engineers and researchers to join the Cloud AI Platform team, delivering data systems and compute for next‑gen ML workloads and GenAI applications across Apple products.

You’ll help scale data pipelines, embeddings, and model workflows, from experimentation to production. Ideal candidates have strong ML foundations, experience with large distributed systems, and proficiency in Python, Java or Go, plus a track record in real‑world ML infrastructure and