Apple logo
Apple Verified
Consumer Electronics, Software, Services, Retail

Machine Learning Engineer - Video Generation Models

CaliforniaOnsiteFull TimePosted 1 month agoVisa sponsorship available

Is this role right for you?

Upload your resume and get a skill-by-skill breakdown — see exactly where you match, where you're close, and what to highlight. Not a mystery percentage.

Get a tailored resume highlighting what this role needs.

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.

We are hiring a machine learning engineer with deep, hands-on experience training large generative models to help build our video generation models. You will work across pre-training, fine-tuning, and inference optimization, from designing the training recipe and running large distributed training jobs through making the resulting models efficient to run. As a member of the team, you will develop fundamental model capabilities and collaborate with engineers and researchers across Apple to advance our products.

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.

Ready to apply?
You'll be redirected to Apple's application page.

Similar roles

See how RiseMe works

Match, tailor, and prep so you send fewer applications and land more interviews.