Video-Generation ML Engineer — Scale & Inference Expert

Apple; San Diego (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Apple is seeking a machine learning engineer to advance video generation models, with hands-on experience in training large generative models from pre-training to fine-tuning and efficient inference.

You will design training recipes, run large distributed training jobs across multi-node GPUs, and optimize both training and inference for production-scale deployment, collaborating with engineers and researchers across Apple.

Qualifications

• Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering, or related field with at least 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 Python programming skills and proficiency in PyTorch or JAX.

Responsibilities

• Pre-training video generation models, including architecture selection and training recipe design.
• Running, monitoring, and debugging large-scale distributed training jobs across multi-node GPU clusters.
• Improving training and inference efficiency through optimization techniques such as parallelism, mixed precision, checkpointing, and hardware utilization.
• Collaborating with product and research stakeholders to translate requirements into modeling and engineering tasks.

Skills

Python programming

Distributed training

Education

Bachelor's degree in EE/CS/CE

Tools

PyTorch

JAX

Job description

Apple is seeking a machine learning engineer to advance video generation models, with hands-on experience in training large generative models from pre-training to fine-tuning and efficient inference.

You will design training recipes, run large distributed training jobs across multi-node GPUs, and optimize both training and inference for production-scale deployment, collaborating with engineers and researchers across Apple.