Staff Machine Learning Engineer — Commerce & Growth Intelligence

Apple, Inc.; San Francisco (CA); On-site; USD 216,000 - 325,000; Full time

Benefits offered by this job

Stock programs

Relocation assistance

Education reimbursement

Medical and dental coverage

Job summary

Apple, Inc. in San Francisco, CA, seeks a Staff Machine Learning Engineer to build inference systems and shared ML infrastructure for customer‑facing products. You will own architectural decisions and deliver across teams, with opportunities to extend into training and device‑to‑cloud integration.

You will lead projects spanning model inference, data processing, and deployment, while mentoring engineers and aligning cross‑functional teams for scalable production systems.

Qualifications

• Bachelor’s degree in CS/CE/IS/EE with 12+ years of relevant work experience.
• Experience shipping production ML systems with ownership of architecture.
• Strong software engineering including API design, testing, and debugging.
• Experience integrating ML models into software systems and deployment.
• Demonstrated history leading technical initiatives across teams.

Responsibilities

• Build inference systems for product workloads and APIs; optimize latency and throughput.
• Develop shared ML infrastructure across models and apps.
• Scale fine-tuning and training workflows with distributed job execution.
• Build data-processing workflows to prepare large training inputs.
• Connect ML systems with product software for on‑device and cloud use.
• Set technical direction, align interfaces, and lead prototyping to production.
• Mentor engineers through design reviews and implementations.

Skills

Production ML systems

Model inference

Distributed computing

System design

Leadership

Communication

Python

Education

Bachelor’s degree in CS/CE/IS/EE

Master’s degree in CS/CE/IS/EE

Tools

Ray

Ray Serve

ONNX Runtime

Core ML

Spark

Flink

LangGraph/MCP

Job description

San Francisco, California, United States Machine Learning and AI

The Commerce & Growth Intelligence team is looking for a Staff Machine Learning Engineer to build the systems that bring ML capabilities into customer-facing products.Our work spans model inference, distributed ML workloads, data processing, and integration with cloud services and on device intelligence on Apple devices. You’ll help shape how these components fit together and build the software that connects them—from making a model available through a reliable inference service to enabling a fine‑tuning workflow to handle larger datasets and workloads. Your primary focus will be model inference and shared ML infrastructure, with opportunities to extend that work into training, fine‑tuning, data processing, and device-to-cloud integration.This is a hands‑on technical leadership role with ownership of architectural decisions and delivery across teams.

Description

You’ll lead engineering projects across the AI/ML stack, working with modeling, infrastructure, and product teams to turn emerging capabilities into production software. Modeling partners lead model architecture and quality evaluation; you’ll lead the engineering systems and integrations that support training, deployment, and product use, while collaborating on model iteration and fine‑tuning. You’ll combine depth in a core area with the ability to work across system boundaries. The work also includes choosing and extending open‑source infrastructure, designing interfaces between components, and resolving practical constraints involving performance, reliability, compute resources, and deployment environments.

Responsibilities

• Build inference systems for product workloads. Develop services and APIs for model execution, including request routing, batching, concurrency, and scaling. Make tradeoffs among latency, throughput, memory use, and compute cost.
• Develop shared ML infrastructure. Adopt and extend open‑source and Apple‑developed technologies where appropriate, to build reusable components across models and applications.
• Scale fine‑tuning and training workflows. Partner with modeling teams to improve distributed job execution, data loading, checkpointing, failure recovery, and accelerator utilization. Responsibilities may also include iterating on and fine‑tuning foundation models and custom‑built models.
• Build data‑processing workflows for ML. Use distributed processing to join, filter, deduplicate, and transform large datasets into reproducible inputs for training and batch inference. Address bottlenecks in data movement and processing.
• Connect ML systems with product software. Design interfaces and implement integrations across inference services, backend applications, and on‑device components, working with client engineers on iOS and device‑to‑cloud workflows.
• Set technical direction and deliver it. Turn ambiguous requirements into architecture and implementation plans, align teams on interfaces and ownership, and lead projects from prototyping through production.
• Raise engineering quality through hands‑on leadership. Build critical components, diagnose issues across data, model execution, and application boundaries, and mentor engineers through design reviews and implementation.

Minimum Qualifications

• Experience shipping production ML systems or ML‑powered applications, with ownership of significant architecture and implementation decisions.
• Deep expertise in at least one relevant area, such as model inference, distributed computing, ML infrastructure, or data systems for ML.
• Strong programming and software engineering skills, including system design, API design, testing, and debugging complex software.
• Practical experience integrating ML models into software systems and understanding their execution and deployment requirements.
• A track record of leading technical initiatives across teams, resolving architectural tradeoffs, and delivering working systems.
• Experience investigating performance and reliability problems and driving improvements based on measured results.
• Ability to communicate technical decisions clearly, build alignment across disciplines.
• Bachelor’s degree in Computer Science, Computer Engineering, Information Systems, or Electronic Engineering and 12+ years of relevant work experience.

Preferred Qualifications

• Distributed ML execution and model serving using Ray, Ray Serve, or comparable technologies.
• Model optimization and deployment using inference runtimes such as ONNX Runtime, Core ML, or comparable systems.
• Distributed training or fine‑tuning, including job orchestration, checkpointing, and GPU or other accelerator workloads.
• Large‑scale data processing with Spark, Flink, or comparable tools, including dataset preparation for ML.
• Go for backend services and inference APIs.
• Swift for iOS services, daemons, or on‑device ML integration.
• Experience with Python for ML workflows, data processing, tooling, or backend services; experience applying Python to agent workflows and tool integration using LangGraph, Model Context Protocol (MCP), or related technologies is an additional plus.
• Applications that combine on‑device capabilities with cloud‑based models and services.
• Master’s degree in Computer Science, Computer Engineering, Information Systems, or Electronic Engineering and 8+ years of relevant work experience.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $216,200 and $324,800, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation.

Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation.

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

Apple accepts applications to this posting on an ongoing basis.