Staff ML Platform Engineer — GenAI & Scale

Apple; Seattle (WA); On-site; USD 180,000 - 250,000; Full time

Job summary

Apple Cloud AI Platform team enables Apple's next generation of intelligent products by giving Apple's ML engineers and researchers the data systems and large-scale compute they need to build and ship models at Apple's bar for quality and privacy.

The Apple Cloud AI Platform team enables Apple's next generation of intelligent products by giving Apple's ML engineers and researchers the data systems and large-scale compute they need to build and ship models at Apple's bar for quality and privacy.

Qualifications

• End-to-end ML workflow experience from data prep to deployment.
• Experience building large-scale distributed systems in production.
• Familiar with generative AI techniques and model optimization.
• Proficient in Java, Python, or Go; strong collaboration and communication.
• Experience configuring and troubleshooting large production environments.

Responsibilities

• Design and build platform behind Apple's largest model builds including data ingestion and governance.
• Develop Python SDKs and data libraries for ML engineers to access datasets.
• Build high-throughput data access for large GPU fleets and distributed pipelines.
• Operate data pipelines with Spark, Daft, and Rust-based systems.
• Ensure tight integration with PyTorch, JAX, and TensorFlow in model development.

Skills

Machine Learning

Distributed Systems

Generative Techniques

Python

Java

Go

Collaboration

Education

B.S./M.S./Ph.D. in Computer Science or Computer Engineering

Tools

Spark

Daft

Rust

PyTorch

JAX

TensorFlow

Docker

Kubernetes

Job description

Apple Cloud AI Platform team enables Apple's next generation of intelligent products by giving Apple's ML engineers and researchers the data systems and large-scale compute they need to build and ship models at Apple's bar for quality and privacy.

The Apple Cloud AI Platform team enables Apple's next generation of intelligent products by giving Apple's ML engineers and researchers the data systems and large-scale compute they need to build and ship models at Apple's bar for quality and privacy.