Staff Machine Learning Engineer

What you'll do

• Design and build platform for versioning and governance of petabyte-scale datasets.
• Develop Python SDKs and core data libraries for ML engineers to access model-ready datasets.
• Build high-throughput data access primitives to feed Apple's largest GPU fleets efficiently.
• Operate distributed data pipelines using Spark, Daft, and Rust-based systems for large-scale data preparation.
• Ensure legal terms of use enforcement, privacy controls, and end-to-end data lineage in platform capabilities.

What we're looking for

• Strong foundation in machine learning with hands-on experience across data preparation, pipeline development, experimentation, evaluation, and deployment.
• Expertise in building and running large-scale distributed systems for ML infrastructure.
• Familiarity with modern generative techniques like transformers, diffusion models, and retrieval-augmented generation.
• Proven experience delivering data and ML infrastructure in real-world production environments.
• Experience configuring, deploying, and troubleshooting large-scale production environments.
• Extensive programming experience in Java, Python, or Go.
• Strong collaboration and communication skills with a comfort level navigating ambiguity.