On-Device ML Integration Engineer, Graphics, Games and Machine Learning

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 $211k

This role $250k

$143k most similar roles pay here $337k

This role pays more than 86% of similar roles. Most pay $184,575–$237,875 — the shaded band above. At the midpoint, this role pays about $250k versus about $211k 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 Integration Engineer, Graphics, Games and Machine Learning

Apply Now Log in to save

As a senior On-Device ML Integration Engineer in the Graphics, Games and Machine Learning team at Apple, you will play a pivotal role in advancing the company’s machine learning capabilities by ensuring seamless integration of cutting-edge models across various devices. Your day-to-day responsibilities include optimizing ML workflows for efficient execution, developing custom kernels and compiler optimizations, and creating robust benchmarking tools to validate performance. You will work closely with model developers and runtime engineers to enhance user experience and functionality while leveraging Apple’s ML infrastructure. This role requires expertise in C++, proficiency in frameworks like PyTorch or JAX, and a deep understanding of GPU programming paradigms and ML architectures such as Transformers. Your contributions will directly impact the performance and reliability of machine learning models across Apple’s diverse hardware ecosystem.

Skills

C++ PyTorch MLX JAX GPU programming MLIR LLVM Hugging Face Swift CI/CD

What you'll do

Ensure functional and performant integration of Apple’s ML models across the inference stack.
Integrate Apple’s ML tools into internal and external model repositories for efficient testing.
Develop optimizations including custom operations and compiler tweaks to enhance inference efficiency.
Spearhead the integration of cutting-edge ML models with peak performance, validating Apple’s inference stack.
Create comprehensive benchmarking and debugging toolchains for ML workflows.

What we're looking for

Bachelor’s degree in Computer Science, Engineering, or related field required.
Proficiency in C++ and at least one ML authoring framework like PyTorch, MLX, JAX.
Deep understanding of machine learning fundamentals and common architectures.
Experience with GPU programming paradigms and optimization techniques.
Strong communication skills for cross-functional collaboration.