Senior Machine Learning Research Engineer, AIML

Apple Inc

Confirmed live 2 days ago Trusted

Quick summary

Work type
On-site
Location
Seattle, WA
Salary
$175,000–$308,500 / yr
Posted
98 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $226k
This role $242k
$159k most similar roles pay here $325k

This role pays more than 57% of similar roles. Most pay $197,631–$254,750 — the shaded band above. At the midpoint, this role pays about $242k 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

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, improving scalability, and creating new techniques to overcome system constraints while potentially open-sourcing your contributions. The role requires expertise in Python, PyTorch internals, and high-performance ML infrastructure. You will apply knowledge of software-hardware co-design principles, LLM architectures, and parallelization algorithms for large model training. This position focuses on the technical challenges of accelerating inference speed and advancing centralized machine learning platforms to support multimodal foundation models and personalized intelligent experiences.

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 while improving the scalability of deep learning models.
  • Create new techniques to overcome existing system constraints in high-performance ML infrastructure.
  • Implement PyTorch-focused optimizations to drive system-level efficiency for large-scale models.
  • Develop and maintain state-of-the-art machine learning frameworks and tools.
  • Build and optimize model architectures for on-device inference.
  • Contribute to the open-source community by releasing research and infrastructure improvements.

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.

More like this

Similar roles

Senior Machine Learning Research Engineer, AIML

Apple Inc

Seattle, WA 148 days ago $175,000$308,500
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI Multimodal Foundation Models On-device Inference ML Infrastructure software–hardware co-design Parallelization Algorithms Machine Learning Research
3+ yrs exp

Senior Machine Learning Research Engineer, AIML

Apple Inc

Seattle, WA 58 days ago $175,000$308,500
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI Machine Learning On-device Inference software–hardware co-design Foundation Models ML Infrastructure
3+ yrs exp

Senior Machine Learning Research Engineer, AIML

Apple Inc

Cupertino, CA 148 days ago $184,700$324,800
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI On-device Inference ML Infrastructure software–hardware co-design Parallelization Algorithms Foundation Models Machine Learning Computational Science
3+ yrs exp

Senior Machine Learning Research Engineer, AIML

Apple Inc

Cupertino, CA 58 days ago $184,700$324,800
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
3+ yrs exp

Senior Machine Learning Research Engineer, AIML

Apple Inc

Cupertino, CA 98 days ago $184,700$324,800
PyTorch Python LLM DeepSpeed FairScale torch.distributed Generative AI Machine Learning On-device Inference software–hardware co-design Scalable ML Infrastructure Foundation Models
3+ yrs exp

Machine Learning Engineer

Apple Inc

Seattle, WA 180 days ago $142,300$263,300
Python PyTorch Deep Learning Computer Vision Object Detection Facial Recognition Data Pipelines Model Training On-device Inference HW/SW Co-design Machine Learning Error Analysis