Video-Generation ML Engineer: Large-Scale Training Expert

Apple Inc.; San Diego (CA); On-site; USD 142,000 - 263,000; Full time

Job summary

Apple Inc. in San Diego, CA is seeking a machine learning engineer to advance video generation through full lifecycle training, from pre-training to fine-tuning and efficient inference. You will design training recipes, run distributed multi-node GPU training, and optimize models for production-ready performance across Apple products.

The role emphasizes collaboration with engineers and researchers across Apple, with a focus on reproducibility, scalability, and impact on video applications.

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.

Responsibilities

• Pre-training video generation models, including architecture selection, training recipe design, hyperparameter and scaling decisions, and evaluation of model quality
• Running, monitoring, and debugging large-scale distributed training jobs, and diagnosing loss instabilities, divergence, and throughput regressions
• Improving training efficiency and cost, including parallelism strategy, mixed-precision training, checkpointing, and hardware utilization
• Improving inference efficiency through step distillation, few-step sampling, and quantization, and characterizing the resulting quality and latency trade-offs
• Adapting pre-trained models through fine-tuning, preference optimization, and knowledge distillation
• Building and maintaining the training and evaluation code the team depends on, with an emphasis on reproducibility and reliability
• Partnering with product and research stakeholders to translate requirements into modeling and engineering tasks

Skills

Python

Distributed training

Video generation

Generative model training

Education

Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering

Tools

PyTorch

JAX

Job description

Apple Inc. in San Diego, CA is seeking a machine learning engineer to advance video generation through full lifecycle training, from pre-training to fine-tuning and efficient inference. You will design training recipes, run distributed multi-node GPU training, and optimize models for production-ready performance across Apple products.

The role emphasizes collaboration with engineers and researchers across Apple, with a focus on reproducibility, scalability, and impact on video applications.