Machine Learning Engineer, Video Generation Models

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
San Francisco, CA
Salary
$150,400–$277,600 / yr
Posted
3 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $225k
This role $214k
$135k most similar roles pay here $293k

This role pays more than 52% of similar roles. Most pay $195,150–$254,750 — the shaded band above. At the midpoint, this role pays about $214k versus about $225k for comparable roles.

Based on 240 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 2000 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1613 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Machine Learning Engineer, Video Generation Models

Machine Learning Engineer - Video Generation Models will join a fast-paced team to develop and optimize video generation models through the entire lifecycle of pre-training, fine-tuning, and inference optimization. The role involves designing training recipes, selecting architectures, managing hyperparameter scaling, and running large distributed training jobs across multi-node GPU clusters. You will be responsible for improving training efficiency via parallelism strategies and mixed-precision training while enhancing inference performance through step distillation, few-step sampling, and quantization. Key responsibilities include building reliable evaluation code, diagnosing loss instabilities, and collaborating with researchers to translate requirements into technical tasks. The ideal candidate possesses strong software engineering skills in Python and proficiency in PyTorch or JAX. Experience with diffusion or autoregressive models, temporal consistency, and long-horizon generation is highly valued for this specific product area.

What does a Machine Learning Engineer earn in California?

Median $246150 from 178 postings across 30 companies.

See salary data

What you'll do

  • Design training recipes, architectures, and hyperparameter scaling for video generation models.
  • Run, monitor, and debug large-scale distributed training jobs across multi-node GPU clusters.
  • Optimize training efficiency through parallelism strategies, mixed-precision training, and improved hardware utilization.
  • Improve inference performance using techniques like step distillation, few-step sampling, and quantization.
  • Adapt pre-trained models through fine-tuning, preference optimization, and knowledge distillation.
  • Build and maintain reliable, reproducible training and evaluation codebases.
  • Translate product and research requirements into specific modeling and engineering tasks.

What we're looking for

  • Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering, or a relevant field.
  • 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 in PyTorch or JAX.
  • MS or PhD in Electrical Engineering, Computer Science, or Computer Engineering (preferred).
  • Experience with video generation architectures including diffusion models, autoregressive models, and temporal consistency (preferred).
  • Experience with large-scale training operations, inference optimization techniques, and model fine-tuning (preferred).

More like this

Similar roles

Machine Learning Engineer, Video Generation Models

Apple Inc

San Diego, CA 3 days ago $142,300$263,300
Python PyTorch JAX Distributed Training Generative Models Autoregressive Models Quantization Knowledge Distillation Fine-tuning Preference Optimization Mixed-precision Training Inference Optimization GPU Clusters Scaling Laws
3+ yrs exp

Machine Learning Engineer, Video Engineering

Apple Inc

Sunnyvale, CA 108 days ago $184,700$324,800
Machine Learning Computer Vision Generative AI Python PyTorch C++ Diffusion Models 3D Reconstruction Neural Radiance Fields 3D Gaussian Splatting Differentiable Rendering Distributed Training On-device ML
10+ yrs exp

Machine Learning Video Engineer

Apple Inc

Cupertino, CA 151 days ago $150,400$277,600
Machine Learning Deep Learning PyTorch TensorFlow Computer Vision Video Processing Image Processing CPU GPU Apple Neural Engine Computer Architecture
3+ yrs exp

Machine Learning Video Processing Engineer

Apple Inc

Cupertino, CA 151 days ago $129,300$194,700
Machine Learning Deep Learning PyTorch TensorFlow Computer Vision Video Processing Image Processing CPU GPU Apple Neural Engine Computer Architecture