Senior ML Engineer, Apple Ray, Apple Data Platform
Core
Design and build core components of Apple's unified data+ML platform powered by open-source Ray to support large-scale training and inference workloads.
Role type
Senior IC distributed systems engineer (ML platform)
Builds
Distributed runtime, orchestration layer, and system APIs for Apple's intelligent features
Domain
Cloud infrastructure + Machine Learning
Deliverable
production ML models
Required skills
distributed systems, Python, C++/Rust/Go/Java, PyTorch/TensorFlow, GPU training, Kubernetes, cluster orchestration, debugging
Preferred skills
DeepSpeed/Horovod/FSDP/ZeRO, GPU workload optimization, ML platform orchestration, open-source contributions, Spark/Flink
Technologies
Ray, Kubernetes, EKS, GKE, PyTorch, TensorFlow, DeepSpeed, Horovod, FSDP, ZeRO, Spark, Flink
Responsibilities
Build scalable distributed systems and platform components using Ray; Develop APIs and services for ML training/inference pipelines; Optimize performance across GPU/CPU clusters; Collaborate with ML teams to translate needs into platform features; Design fault-tolerant orchestration and autoscaling; Diagnose complex distributed system issues; Improve observability and monitoring for ML workloads.
Seniority
Senior, hands-on IC