Senior ML Infrastructure Engineer
Core
Design and own the architecture of Apple's large-scale ML data platform, enabling reproducible model training and inference at petabyte scale.
Role type
Staff ML Infrastructure Engineer (Platform Architecture)
Builds
Data ingestion, versioning, lineage, and high-throughput delivery systems for GPU/TPU training fleets
Domain
Generative AI, Large-scale Distributed Data Systems, ML Infrastructure
Deliverable
production ML models
Required skills
Distributed data systems, Python, Rust, C++/Go, I/O-bound performance engineering, Columnar/lakehouse formats (Parquet, Iceberg, Delta), ML workflow architecture, System design, Technical leadership
Preferred skills
PyTorch/JAX/TensorFlow data-loading layers, Ray Data/NVIDIA DALI, GPU/TPU fleet saturation, Data lineage/governance (DataHub, Unity Catalog), Spark/Daft/Polars internals, Kubernetes
Technologies
Python, Rust, C++, Go, Parquet, Iceberg, Delta, Lance, PyTorch, JAX, TensorFlow, Ray Data, NVIDIA DALI, WebDataset, Spark, Kubernetes, Docker
