Staff Machine Learning Engineer, Tools and Framework AI

Apple Inc

Quick summary

Work type
On-site
Location
Cupertino, CA
Salary
$181,100–$318,400 / yr
Posted
96 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $222k
This role $250k
$154k most similar roles pay here $336k

This role pays more than 75% of similar roles. Most pay $193,825–$250,250 — the shaded band above. At the midpoint, this role pays about $250k versus about $222k 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 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 · Staff Machine Learning Engineer, Tools and Framework AI

As a Staff Machine Learning Engineer on the Tools and Frameworks AI team, you will lead the development of cutting-edge AI solutions that enhance software productivity. Your role involves architecting and implementing machine learning systems from concept to production, ensuring they meet real-world demands with high-quality code. You will collaborate closely with product managers, researchers, and engineers across Apple to turn ambitious ideas into robust products, making critical design decisions along the way. Key responsibilities include mentoring junior team members, driving requirements with cross-functional stakeholders, and delivering innovative solutions that significantly impact user experience. The ideal candidate has extensive experience in machine learning modeling, strong programming skills in Python, Java, C++, or Swift, and a proven track record of solving complex problems creatively.

What you'll do

  • Lead the design and implementation of AI-first developer productivity projects from prototype to production.
  • Ensure end-to-end user experience quality by establishing best software engineering practices.
  • Collaborate cross-functionally with stakeholders to drive requirements and deliver solutions.
  • Mentor junior engineers, enhancing their skills and productivity within the team.
  • Make critical technical decisions that guide the transition of exploratory work into robust production systems.

What we're looking for

  • At least 8 years of hands-on experience in machine learning and software engineering.
  • Expert knowledge in machine learning modeling and applied research.
  • Proficient programming skills in Python, Java, C++, or Swift.
  • Proven ability to deliver high-quality products and solve complex problems innovatively.
  • Strong collaboration and communication skills for working with cross-functional teams.
  • Experience leading AI/ML projects and mentoring junior engineers.
  • Hands-on experience building ML infrastructure at scale.

More like this

Similar roles

Staff Machine Learning Engineer

Apple Inc

Cupertino, CA 48 days ago $212,000$318,400
Python PyTorch JAX TensorFlow Spark Daft Rust Java Go Kubernetes Docker CI/CD Parquet Iceberg Delta Lance Ray Data NVIDIA DALI WebDataset Mosaic StreamingDataset Arrow DataHub OpenLineage Unity Catalog Polars DuckDB