Senior ML Platform Engineer – Scale Training & Inference

Apple, Inc.; New York (NY); On-site; USD 263,000 - 394,000; Full time

Benefits offered by this job

Stock programs

Employee stock purchase plan

Education reimbursement

Job summary

Apple Inc. in New York City is seeking a Senior Staff Machine Learning Engineer for the ML Platform.

You will design and build platform capabilities to enable Apple Ads teams to train, fine-tune, and deploy ML models at scale while upholding Apple’s privacy commitments. You will collaborate with ML engineers and data scientists to create reliable training pipelines and inference services, selecting architectures and tooling to scale across teams.

Qualifications

• Experience building shared ML platforms used by multiple teams.
• Deep understanding of the ML lifecycle, including training pipelines, evaluation methodologies, and deployment patterns.
• Deep understanding of deep learning architectures (Transformers, LLMs, DNNs) and training frameworks (TensorFlow, PyTorch).
• Prior experience applying ML at scale in Ads, recommender systems, information retrieval or related domains.
• Prior experience in distributed training at scale and optimization techniques like pruning, compression, quantization & distillation.
• Ability to communicate effectively with technical and non-technical teams.
• Results oriented with strong technical leadership and ownership mindset.
• Curious business attitude with drive to own projects.

Responsibilities

• Design and develop model training and fine-tuning infrastructure at scale.
• Build and operate high-performing ML platform capabilities across teams.
• Collaborate with ML engineers and scientists to improve features, models, and applications.
• Define architectures that meet unique advertising network challenges and privacy commitments.

Skills

ML platforms

ML lifecycle

Transformers

TensorFlow

PyTorch

Distributed training

Communication

Leadership

Ads domain

Education

PhD in Computer Science

MS in Computer Science

BS in Computer Science

Tools

TensorFlow

PyTorch

Airflow

Kubernetes

Job description

Apple Inc. in New York City is seeking a Senior Staff Machine Learning Engineer for the ML Platform.

You will design and build platform capabilities to enable Apple Ads teams to train, fine-tune, and deploy ML models at scale while upholding Apple’s privacy commitments. You will collaborate with ML engineers and data scientists to create reliable training pipelines and inference services, selecting architectures and tooling to scale across teams.