Apple
Senior ML Infrastructure Engineer - VE Algorithms
$142K–$263K
San Diego, California · Senior · Full-time · Posted Apr 21, 2026
- Experience
- Minimum 3 years industry experience.
- Education
- BS in Electrical Engineering/Computer Science or a related field, with a focus on machine learning.
About the job
Build and optimize infrastructure for training, adapting, and deploying large-scale generative models for video and image features.
What you'll do
- Optimize and profile visual pre-training pipelines across diffusion and auto-regressive architectures.
- Design workload scheduling for distributed training across clusters of thousands of GPUs and optimize the compilation stack.
- Build experimentation tools, dashboards, and scalable CI/CD pipelines for multi-GPU workflows.
What they ask for
- BS in Electrical Engineering, Computer Science, or a related field, with a focus on machine learning.
- At least 3 years of industry experience.
- Experience training and adapting LLMs and with distributed training of large models.
- Advanced fluency in PyTorch; excellent programming skills in Python or C++ and experience contributing software to large projects.
Nice to have
- Strong Machine Learning fundamentals.
- Experience working with large cross-functional and diverse teams.
Pay
$142,300 and $263,300 USD base salary
Tools
PyTorch, Python, C++, diffusion, auto-regressive architectures, GPUs, CI/CD
AI focus
Infrastructure for training, adapting, and deploying large-scale generative models; distributed visual pre-training for images and videos.
Model-written summary · the original posting is authoritative