Senior ML Production Orchestrator for On-Device AI

Apple Inc.; Cupertino (CA); On-site; USD 181,100 - 318,400; Full time

Benefits offered by this job

Comprehensive medical and dental coverage

Retirement benefits

Employee stock purchase plan

Job summary

Apple Inc. in Cupertino, California is looking for an experienced professional to lead the model lifecycle in Machine Learning and AI. This role involves owning model pipelines and integrating them with Apple frameworks to ensure efficient deployments and updates across various platforms.

The ideal candidate will possess strong software engineering skills, experience with cloud ML platforms, and a solid understanding of the ML training lifecycle. This is an opportunity to work at the forefront of AI technology while contributing to innovative product developments.

Qualifications

• 5+ years experience in Machine Learning Operations.
• Production experience with cloud ML platforms including submitting jobs and debugging schedulers.
• Experience with infrastructure-as-code and CLI tool design.

Responsibilities

• Own the end-to-end model lifecycle building model pipelines.
• Design and operate agent-based automation pipelines for ML models.
• Develop multi-agent workflows using LLM-native tooling.

Skills

Software engineering fundamentals

Python

Bash

Machine Learning Operations

Cloud ML platforms

Infrastructure-as-code

Education

Bachelor's degree in Computer Science

Tools

GCP TPU

AWS GPU clusters

Kubernetes

JAX

XLA

Job description

Apple Inc. in Cupertino, California is looking for an experienced professional to lead the model lifecycle in Machine Learning and AI. This role involves owning model pipelines and integrating them with Apple frameworks to ensure efficient deployments and updates across various platforms.

The ideal candidate will possess strong software engineering skills, experience with cloud ML platforms, and a solid understanding of the ML training lifecycle. This is an opportunity to work at the forefront of AI technology while contributing to innovative product developments.