
Machine Learning Engineer - Video Generation Models
Required skills
- Generative Models
- Video Generation
- Distributed Training
- Python
- PyTorch
- JAX
- Inference Optimization
- Fine-tuning
- Pre-training
- Multi-node GPU Clusters
Nice to have
- Diffusion Models
- Autoregressive Models
- Temporal Consistency
- Long-Horizon Generation
- Foundation Models
- Scaling Laws
Role summary
Apple is seeking a Machine Learning Engineer with deep expertise in training large generative models for video generation. The role involves the full model lifecycle, from pre-training and fine-tuning to inference optimization, with a focus on creating reproducible and production-ready solutions. Responsibilities include designing training strategies, managing distributed GPU training, and enhancing model efficiency. The position requires strong Python and deep learning framework skills, with experience in video generation architectures and large-scale training operations being highly desirable.
Description
As a member of our fast-paced group, you'll have the unique and rewarding opportunity to shape upcoming products from Apple. We are looking for someone who has taken large generative models through the full lifecycle, from pre-training through fine-tuning and efficient inference, and can bring that depth to video, with the engineering skills to make that work reproducible and production-ready.
Minimum Qualifications
Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering, or relevant degree, and a minimum of 3 years relevant industry experience Experience with large-scale generative model training for video generation Experience running distributed training across multi-node GPU clusters Strong software engineering skills in Python, with proficiency in a modern deep learning framework such as PyTorch or JAX
Preferred Qualifications
MS or PhD in Electrical Engineering, Computer Science, or Computer Engineering Experience with video generation architectures, including diffusion or autoregressive models, temporal consistency, and long-horizon generation Experience contributing to major foundation or base model pre-training efforts, including scaling laws and transferring training recipes across model and training scales Experience with large-scale training operations, including parallelism strategies and diagnosing loss instability, divergence, or throughput regressions Experience improving and adapting trained models, such as step distillation, few-step sampling, or quantization for inference efficiency, and supervised fine-tuning, preference optimization, or knowledge distillation for quality Ability to work through ambiguity, collaborate across teams and disciplines, and communicate complex technical results clearly
Sample Apple interview questions
- 1
Design a typeahead box for a search engine
system designmedium - 2
Design a rate limiter
system designmedium - 3
Architect a key-value store for distributed systems.
system designmedium - 4
Design a system to schedule jobs in a distributed environment
system designmedium - 5
What is an API and how does it work?
technicalmedium
Get started to unlock a personalized interview prep pack for this role.
Similar roles
HI Production Designer - Systems Design - Apple Media ServicesApple · Culver City, CA · Onsite
Sr. Software Engineer (Java Middleware), eBusinessApple · Sunnyvale, CA · Onsite- Senior Machine Learning EngineerCheckr · Remote · Remote
Staff Machine Learning Engineer - Team Lead (Active Secret Clearance)Striveworks · Remote · Remote- Machine Learning Engineer - New GradSambaNova Systems · Austin, TX · Onsite
- Principal Machine Learning EngineerPhysicsX · New York, NY · Onsite
