AIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & Technology
Core
Building and maintaining compute infrastructure to run large-scale ML workloads (training, inferencing, fine-tuning) at Apple, ensuring stability, reliability, efficiency, and cost-effectiveness.
Role type
Senior IC machine learning infrastructure engineer (ML compute & platform)
Builds
Scalable ML compute infrastructure and platforms for training and inference
Domain
Cloud infrastructure + Machine Learning
Deliverable
infrastructure
Required skills
distributed systems, containerization, cloud platforms, Python, Go, Kubernetes, Ray, Beam, Flink, debugging accelerators (GPU, TPU)
Preferred skills
ML training/deployment frameworks (JAX, Tensorflow, PyTorch, TensorRT, vLLM), advanced degrees in CS/engineering
Technologies
Kubernetes, Ray, Beam, Flink, JAX, Tensorflow, PyTorch, TensorRT, vLLM, GPU, TPU, AWS Trainium
Responsibilities
Scope, architect, and deliver innovative high-quality solutions; Collaborate with teams across Apple on ML workloads; Conduct code reviews; Help onboard new team members and provide mentorship
Seniority
Senior, hands-on IC
