← All roles

Senior Machine Learning Engineer

Apple· Software and Services
Apply on company site
Seattle, WA$175k - $308.5kFull-timePosted 1mo ago

About the role

Join a team at the forefront of ML infrastructure and generative AI, building robust systems that connect scalable data pipelines with advanced ML workflows to enable next-generation intelligent experiences on Apple products and services. The role focuses on ML data and feature platforms, embeddings, AI Ops, efficient inference, and a modern feature platform to streamline experimentation and drive innovation. You’ll work on a platform that supports GenAI workloads with strong emphasis on data-centric ML, reproducibility, and scalable infrastructure across large-scale model training and deployment.

Responsibilities

  • Build and own high-throughput data access and loading primitives that feed Apple's largest GPU and TPU fleets, keeping training compute-bound rather than I/O-bound.
  • Develop and evolve the Python and Rust data libraries and SDKs that ML engineers depend on to access, transform, and load model-ready datasets across every stage of model development.
  • Build and operate distributed data pipelines across Spark, Daft, and Rust-based systems for ingestion, transformation, and large-scale data preparation.
  • Contribute to the platform behind Apple's largest model builds: ingestion, immutable versioning, lineage, and governance across structured, unstructured, and multimodal data at petabyte scale, so every model run is reproducible from a versioned dataset.
  • Make dataset access a first-class concern in the model development loop through tight integration with the data-loading layers of PyTorch, JAX, and TensorFlow.
  • Partner with research and product teams to onboard new data sources and unblock fast iteration on the datasets powering foundation-model and GenAI workloads.
  • Diagnose, fix, and automate away complex issues across the stack, from ingestion pipelines to dataset APIs to framework integrations, to maximize uptime and throughput.

Qualifications

  • 5+ years of work experience in machine learning infrastructure, distributed data systems, or a related field.
  • 5+ years of experience building and shipping data or ML infrastructure and systems in production.
  • Experience operating production systems: monitoring, observability, on-call, and incident response.
  • Strong programming skills in Python, plus a systems language for performance-critical work (Rust strongly preferred; C++ or Go acceptable).
  • Familiarity with columnar and lakehouse formats (Parquet, Iceberg, Delta, or Lance) and the trade-offs between them.
  • Hands-on performance engineering for I/O-bound workloads: Arrow, zero-copy, memory mapping, async I/O, and high-throughput object storage access patterns.
  • Working knowledge of the end-to-end ML workflow and how training and inference workloads consume data, enough to design data systems that serve them well.
  • Familiarity with modern ML and generative techniques (transformers, diffusion, retrieval-augmented generation, fine-tuning) at the level needed to design for those consumers, not necessarily to build the models yourself.
  • Experience designing and operating scalable, highly available systems that prioritize ease of use for the engineers who depend on them.
  • Strong collaboration and written and verbal communication skills.
  • BS, MS, or PhD in Computer Science, Computer Engineering, or equivalent practical experience.

Skills

Required:Machine Learning fundamentalsDistributed systemsGenAI / Generative models

Benefits

  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Discounted products and free services
  • Educational expense reimbursement (tuition)
  • Potential discretionary bonuses or commissions
  • Relocation assistance
  • Apple employee stock programs including RSUs and Employee Stock Purchase Plan
Last updated Sep 25, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple