Senior ML Platform Engineer — Scalable AI Services

Apple; Seattle (WA); On-site; USD 150,000 - 190,000; Full time

Job summary

Apple is seeking an ML Engineer to build and operate managed platform services at scale within the ASE organization. You will own end-to-end ML workflows, embedding and retrieval pipelines, feature stores, and governance across production AI workloads.

You will partner with teams across Apple to deploy production solutions, optimize costs, and ensure reliable on-call operations. This role demands strong Python/Rust/Java skills and extensive distributed systems experience.

Qualifications

• 3+ years building production ML systems or ML infrastructure.
• Strong programming skills in Python and/or Rust/Java.
• Understanding end-to-end ML workflows from data prep to deployment.
• Experience with distributed data processing and large-scale systems.
• Experience with model serving and inference optimization.
• Experience building APIs and services.
• Strong collaboration and communication skills.
• Comfortable navigating ambiguity in fast-moving areas.

Responsibilities

• Design, build, and optimize large-scale ML platform services used across Apple.
• Develop embedding and retrieval path end-to-end including vector indexes and quality evaluation.
• Build and operate the feature store used for training and serving.
• Improve ML workloads cost and quality via routing, caching, and configuration.
• Create self-service experiences enabling data-to-production AI.
• Support training with supervised fine-tuning and RL methods.
• Develop governance, lineage, and access control for ML workloads.
• Collaborate with customer teams across Apple to deliver production solutions.
• Operate production services with on-call responsibilities.

Skills

Production ML

Python/Rust/Java

ML workflows

Distributed systems

Model serving

APIs

Collaboration

Ambiguity

Education

BS/MS/PhD in CS or equivalent

Tools

vLLM

TensorRT

Ray Serve

Spark

Kubernetes

AWS/GCP

Job description

Apple is seeking an ML Engineer to build and operate managed platform services at scale within the ASE organization. You will own end-to-end ML workflows, embedding and retrieval pipelines, feature stores, and governance across production AI workloads.

You will partner with teams across Apple to deploy production solutions, optimize costs, and ensure reliable on-call operations. This role demands strong Python/Rust/Java skills and extensive distributed systems experience.