ML Framework Engineer, Graphics, Game and ML
Apple Inc
Quick summary
Market check
How this pay compares to similar roles
This role pays less than 80% of similar roles. Most pay $196,562–$254,750 — the shaded band above. At the midpoint, this role pays about $188k versus about $226k for comparable roles.
Based on 240 similar postings.
Employer
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 1984 open roles on FindRole.
Listed pay typically runs $175,000–$277,600 across 1590 roles with salary data.
Most-posted roles
At a glance
The ML Framework (MetalLM) Engineer joins the GPU, Graphics and Machine Learning team to enable high-performance, distributed inference of GenAI applications like LLMs on Private Cloud Compute. You will develop kernel and compiler level optimizations while implementing features for Metal device backends to accelerate training frameworks such as PyTorch and JAX. Day-to-day responsibilities include optimizing code for scalable inference using data, tensor, pipeline, and expert parallelism, alongside applying techniques like quantization, compression, and speculation to improve throughput and reduce latency. The role requires expertise in C/C++/ObjC, GPU kernel development with Metal or CUDA, and system level programming. You will collaborate with hardware and compiler teams to align software performance with Apple Silicon capabilities. Preferred skills include experience with Triton, OpenXLA, LLVM/MLIR, and contributions to major AI frameworks like PyTorch, JAX, or TensorFlow.
Skills
What you'll do
What we're looking for
Related searches
More like this
Apple Inc
Apple Inc
Apple Inc
Apple Inc
Apple Inc
Apple Inc