← All roles

Senior ML Infrastructure Engineer - VE Algorithms

Apple· Machine Learning and AI
Apply on company site
San Diego, CA$142.3k - $263.3kfull-timePosted 6mo ago

About the role

Senior ML Infrastructure Engineer focused on building and optimizing infrastructure for training, adapting, and deploying large-scale generative models for Apple’s Video Engineering (VE) team. Hands-on role emphasizing distributed and parallelized training, profiling, and tooling for multi-GPU workflows in video/image ML pipelines.

Responsibilities

  • Optimize and profile training pipelines for large-scale visual pre-training across diffusion and auto-regressive architectures.
  • Design and implement workload scheduling strategies for distributed training across clusters of thousands of GPUs.
  • Profile and optimize across the full compilation stack — from high level graph capture through intermediate representations down to target HW.
  • Build and maintain experimentation tooling, dashboards, and scalable CI/CD pipelines tailored to multi-GPU training workflows.

Qualifications

  • BS in Electrical Engineering/Computer Science or a related field, with a focus on machine learning and minimum 3 years industry experience.
  • Experience in training and adapting LLMs.
  • Advanced fluency in PyTorch.
  • Excellent programming skills in Python or C++ and experience contributing software to large projects.
  • Experience with distributed training of large models.

Skills

Required:Machine Learning fundamentalsPyTorchPythonDistributed trainingLLMs
Nice to have:C++

Benefits

  • Comprehensive medical and dental coverage
  • retirement benefits
  • discounted products and free services
  • reimbursement for certain educational expenses — including tuition
  • discretionary bonuses or commission payments and relocation (eligible)
  • participation in Apple stock programs (stock purchase plan, RSU potentially)
Last updated Jul 3, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple