On-device ML Infrastructure Engineer, Compiler & Runtime, Graphics, Games & ML

Apple Inc

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$181,100–$318,400 / yr
Posted
56 days ago
Nearby
99+ roles within 25 mi

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $206k
This role $250k
$136k most similar roles pay here $338k

This role pays more than 89% of similar roles. Most pay $175,400–$236,325 — the shaded band above. At the midpoint, this role pays about $250k versus about $206k for comparable roles.

Based on 239 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 1723 open roles on FindRole.

Listed pay typically runs $162,500–$272,100 across 1398 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · On-device ML Infrastructure Engineer, Compiler & Runtime, Graphics, Games & ML

Apply Now Log in to save

As a senior On-device ML Infrastructure Engineer at Apple, you will join a dynamic team bridging research, software engineering, hardware engineering, and product development to advance AI on Apple silicon. Your primary responsibilities include leading critical initiatives for on-device machine learning infrastructure, designing and maintaining systems that power Apple’s ML features, collaborating with hardware compilers and external users to optimize ML models across various devices, and ensuring seamless integration of these tools. You will work extensively with C++, Python, Swift, and other relevant technologies while leveraging expertise in common compiler optimizations and system software engineering. This role demands a deep understanding of machine learning fundamentals, including architectures like Transformers, and familiarity with on-device ML stacks such as TFLite or ONNX, along with experience in open-source models and authoring frameworks like PyTorch or TensorFlow.

Skills

C++ Python Swift TFLite ONNX ExecuTorch Mistral Phi Gemma Huggingface MLIR LLVM TVM PyTorch TensorFlow JAX CUDA OpenGL CI/CD

What you'll do

  • Design and build critical machine learning infrastructure for Apple’s features.
  • Optimize ML models to run efficiently on a variety of Apple devices.
  • Integrate compiler technology with runtime components and kernel libraries.
  • Work with hardware compilers to leverage Apple's ML hardware capabilities.
  • Ensure the infrastructure supports diverse ML models from various developers.

What we're looking for

  • 5+ years of hands-on experience in computer science or engineering.
  • Proficiency in C++ and familiarity with Python and Swift.
  • Understanding of ML fundamentals and common architectures like Transformers.
  • Experience with on-device ML stacks such as TFLite, ONNX, ExecuTorch.
  • Knowledge of open source machine learning models and compiler stacks.

Related searches

More On-device ML Infrastructure Engineer, Compiler & Runtime, Graphics, Games & ML jobs On-device ML Infrastructure Engineer, Compiler & Runtime, Graphics, Games & ML jobs in Cupertino All jobs at Apple Inc

More like this

Similar roles