The point where experts and best companies meet
Apple AIML - Staff ML System Engineer Platform & Technologies MLPT United States, California 730528691
08.06.2025
Share
We're searching for strong machine learning engineers to help build next-generation platform for training deep learning models at scale. You'll be part of a team of distributed systems and machine learning experts, focusing on reliability, scalability and performance of ML related workloads including model training, inference and data processing.
- Experience building large-scale deep learning infrastructure or platforms for distributed model training
- Experience with large-scale AI training infra components, such as accelerators, network fabrics, CUDA, NCCL, RDMA
- Strong programming skills in Python or Go
- Understanding of data structures, software design principles, and algorithms
- Experience building large-scale distributed systems with tools such as Kubernetes, Kafka, Prometheus, etc.
- Experience with deep learning frameworks, such as PyTorch, or JAX
- With minimum of 7+ years of industry experience
- Bachelors degree in the area of Computer Science or equivalent, or a related domain
- Experience working with public cloud vendors such as AWS, GCP, Azure.
- Experience developing model parallel and data parallel training solutions and other training optimizations
- Familiarity with recent developments in foundation model architectures for language and multimodal
- Publication record at ML conferences such as MLSys, NeurIPS, etc.
- Advance degree in the area of Computer Science or equivalent, or a related
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
