2 months ago
USA (city not specified)Mid Level
H1B sponsor
Responsibilities
- Design training recipes and run large distributed training jobs for video generation models.
- Develop video generation model capabilities across pre-training, fine-tuning, and inference optimization.
- Run distributed training across multi-node GPU clusters.
- Improve trained models through techniques such as step distillation, few-step sampling, quantization, supervised fine-tuning, preference optimization, and knowledge distillation.
- Make model development reproducible, production-ready, and efficient to run.
- Collaborate with engineers and researchers across Apple and communicate technical results clearly.
Requirements
- Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering, or a relevant degree.
- Minimum of 3 years of 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 and proficiency with a modern deep learning framework such as PyTorch or JAX.
- Preferred: MS or PhD in Electrical Engineering, Computer Science, or Computer Engineering.
- Preferred: experience with video generation architectures, including diffusion or autoregressive models, temporal consistency, and long-horizon generation.
- Preferred: experience contributing to foundation or base model pre-training, including scaling laws and transferring training recipes across scales.
- Preferred: experience with parallelism strategies and diagnosing loss instability, divergence, or throughput regressions.
Categories
About Apple
Apple designs and sells consumer electronics, software, and services for consumers and professionals worldwide, including iPhone, Mac, iPad, Apple Watch, and AirPods, plus platforms like iOS/macOS and services such as the App Store, iCloud, Music, and TV+. Its business combines device sales with services and subscriptions and in-house silicon design. Founded in 1976, Apple is headquartered in Cupertino, California, and trades on NASDAQ as AAPL.