← All roles

On-Device ML Quality Infrastructure Engineer, Graphics, Games & ML

Apple· On-Device Machine Learning / QA Infrastructure
Apply on company site
Cupertino, CA$184.7k - $324.8kFull-timePosted 2w ago

About the role

Engineer focused on building and maintaining the infrastructure that connects, validates, and benchmarks the On-Device Machine Learning stack at Apple. Owns build and CI systems, automation frameworks, and software quality activities including testing, fuzzing, performance benchmarking, and regression tracking across ML, compiler, and hardware teams.

Responsibilities

  • Design, build, and maintain build systems and CI/CD pipelines for the On-Device ML stack.
  • Develop and maintain automation frameworks for functional, regression, and integration testing.
  • Contribute to robustness and reliability initiatives, including fuzz testing of stack components such as parsers, compilers, and runtimes.
  • Track and investigate performance regressions; work with engineering teams to resolve them.
  • Identify gaps in testing coverage and tooling; drive improvements proactively.
  • Develop and maintain tooling to help engineers diagnose, debug, and fix issues across the ML stack.
  • Collaborate with Apple's CI and infrastructure teams to scale and accelerate validation workflows.
  • Work cross-functionally with ML, compiler, and hardware teams to ensure infrastructure meets their needs.

Qualifications

  • Bachelors in Computer Science or relevant disciplines.
  • Build systems (CMake, Bazel) and CI/CD experience
  • Strong programming and software design skills in Python.
  • In depth knowledge of quality practices and fundamentals, including test planning, automation, and performance evaluation.
  • Excellent collaboration and communication skills.
  • Masters in Computer Science or relevant disciplines.
  • Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.).
  • Experience with standard ML architectures such as Transformers, CNNs or Stable Diffusion a strong plus.
  • Experience with fuzz testing tools or methodologies
  • Experience with performance profiling and analysis tools
  • Familiarity with GenAI tooling for developer productivity or test generation.

Skills

Required:Build systemsCI/CDPythonTesting
Nice to have:Fuzz testingPerformance profilingML frameworks

Benefits

  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Discounted products and free services
  • Educational expense reimbursement including tuition
  • Discretionary bonuses or commission payments
  • Relocation assistance (potential)
  • Employee stock programs including RSU options and Employee Stock Purchase Plan (ESPP)
Last updated Oct 9, 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