Apple

Staff Machine Learning Engineer — Commerce & Growth Intelligence

Apple
Apply
10 hours ago
Cupertino, CA, USA or San Francisco, CA, USAStaff+
H1B sponsor

Responsibilities

  • Lead engineering projects across the AI/ML stack and deliver production software with modeling, infrastructure, and product teams.
  • Build model inference services, shared ML infrastructure, and integrations connecting models to customer-facing products.
  • Enable distributed training, fine-tuning, data processing, deployment, and device-to-cloud workflows.
  • Make architectural decisions, design component interfaces, extend open-source infrastructure, and resolve performance, reliability, compute, and deployment constraints.
  • Investigate performance and reliability issues using measured results and drive improvements.
  • Lead technical initiatives across teams, resolve architectural tradeoffs, and communicate technical decisions clearly.

Requirements

  • Experience shipping production ML systems or ML-powered applications with ownership of significant architecture and implementation decisions.
  • Deep expertise in model inference, distributed computing, ML infrastructure, data systems for ML, or a related area.
  • 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 execution and deployment requirements.
  • Experience leading technical initiatives across teams and delivering working systems.
  • Bachelor’s degree in Computer Science, Computer Engineering, Information Systems, or Electronic Engineering plus 12+ years of relevant work experience.
  • Preferred experience includes Ray or comparable distributed ML systems, ONNX Runtime or Core ML, distributed training and fine-tuning, Spark or Flink, Go, Swift, Python, and on-device/cloud ML integration.
  • A preferred alternative is a master’s degree in a listed technical field plus 8+ years of relevant work experience; experience with LangGraph, Model Context Protocol, or related agent workflow tools is an additional plus.

Tech Stack

Apache FlinkApache SparkGoiOSPythonSwift
Apple

About Apple

10,000+ employees

Apple designs and sells consumer electronics, software, and services for consumers and professionals worldwide, including iPhone, Mac, iPad, Apple Watch, and AirPods, plus platforms like iOS/macOS and services such as the App Store, iCloud, Music, and TV+. Its business combines device sales with services and subscriptions and in-house silicon design. Founded in 1976, Apple is headquartered in Cupertino, California, and trades on NASDAQ as AAPL.

Contact me