Staff ML Engineer: Inference & ML Infra for On-Device Apps

Apple; Cupertino (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Apple in Cupertino seeks a Staff Machine Learning Engineer to build systems that bring ML capabilities into customer-facing products across devices. You will own model inference, distributed ML workloads, data processing, and integration with cloud services and on-device intelligence on Apple devices.

You'll lead engineering projects across the AI/ML stack, guiding model iteration, deployment, and product integration while collaborating with modeling and infrastructure teams.

Qualifications

• Shipping production ML systems with architectural ownership.
• Deep expertise in 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.

Responsibilities

• Lead ML engineering projects spanning model inference, deployment, and infrastructure.
• Collaborate with modeling, data, and product teams to productionize ML capabilities.
• Design interfaces between components and extend open-source infrastructure while optimizing performance and reliability.

Skills

Production ML systems

Model inference

ML infrastructure

Distributed computing

System design

API design

Leadership across teams

Performance & reliability

Communication of decisions

Python for ML workflows

Education

Bachelor's degree in Computer Science/Engineering/IS/EE

Tools

Ray / Ray Serve

ONNX Runtime / Core ML

Spark / Flink

Python for ML

LangGraph / MCP

Swift for iOS

Go for backend services

Distributed training

Job description

Apple in Cupertino seeks a Staff Machine Learning Engineer to build systems that bring ML capabilities into customer-facing products across devices. You will own model inference, distributed ML workloads, data processing, and integration with cloud services and on-device intelligence on Apple devices.

You'll lead engineering projects across the AI/ML stack, guiding model iteration, deployment, and product integration while collaborating with modeling and infrastructure teams.