Senior Machine Learning Research Engineer, AIML

Apple Inc

Confirmed live yesterday Trusted

Quick summary

Work type: On-site
Location: Cupertino, CA
Salary: $184,700–$324,800 / yr
Posted: 58 days ago
Freshness: Confirmed live yesterday
Nearby: 99+ roles within 25 mi

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $226k

This role $255k

$168k most similar roles pay here $342k

This role pays more than 78% of similar roles. Most pay $197,631–$254,750 — the shaded band above. At the midpoint, this role pays about $255k versus about $226k for comparable roles.

Based on 240 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 1984 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1590 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Senior Machine Learning Research Engineer, AIML

Role Posting Log in to save

As a Senior Machine Learning Research Engineer, AIML, you will join the multidisciplinary AIML team within the software organization to develop core components for a scalable machine learning platform. You will design software systems and algorithms that enable performant training and inference for AI-driven experiences across both on-device and server environments. Your daily work involves pushing the limits of existing training technologies, overcoming system constraints, and creating new techniques for large-scale deep learning. The role requires expertise in Python, PyTorch internals, and high-performance ML infrastructure. You will utilize knowledge of LLM architectures, software-hardware co-design principles, and parallelization algorithms like DeepSpeed or FairScale. This position focuses on the technical challenges of accelerating training speeds and improving scalability for multimodal foundation models while potentially contributing to open-source projects within the machine learning domain.

Skills

PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI On-device Inference ML Infrastructure software–hardware co-design Parallelization Algorithms Foundation Models Machine Learning Research

What you'll do

Design software systems and algorithms for performant, scalable training and inference across on-device and server environments.
Develop core components for a centralized, large-scale machine learning platform.
Optimize training and inference speeds by pushing the limits of existing technologies.
Create new techniques to overcome system constraints in deep learning workflows.
Drive system-level efficiency for both small on-device models and large-scale foundation models.
Maintain and improve state-of-the-art machine learning frameworks using the PyTorch stack.
Implement parallelization algorithms for training large-scale models using libraries like DeepSpeed or PyTorch Distributed.
Contribute to open-source projects based on your research and engineering work.

What we're looking for

PhD or Master's degree in Computer Science, or equivalent industry experience.
3+ years of experience in the AI/ML field.
Strong Python programming skills.
Solid understanding of software-hardware co-design principles and algorithms.
Solid understanding of the PyTorch software stack and experience maintaining state-of-the-art ML frameworks.
Strong understanding of LLM architectures and their core building blocks.
Experience with parallel training libraries such as PyTorch Distributed, DeepSpeed, or FairScale.
Experience building ML models for on-device inference.

Machine Learning Engineer

Apple Inc

Cupertino, CA 148 days ago $184,700–$324,800

C++ Python GPU Programming Machine Learning Swift Numerical Computing Distributed Training Software Engineering Low-level Programming MLX Apple Silicon

Save