Sr Staff Machine Learning Engineer, ML Platform

Apple Inc

Actively hiring Posted this week
New York City, NY Posted 4 days ago $212,000–$386,300 / year
View original post Log in to save

At a glance

AI generated

TL;DR

As a Senior Staff Machine Learning Engineer on the ML Platform team at Apple Ads, you will design and develop scalable model training and fine-tuning infrastructure to enable rapid innovation across various teams. Your day-to-day responsibilities include collaborating with other engineers and scientists to build high-performing machine learning systems that adhere to strict privacy standards while delivering cutting-edge advertising solutions. You will leverage deep knowledge of frameworks like TensorFlow and PyTorch, as well as expertise in distributed training and optimization techniques such as model pruning and quantization. This role requires experience in building shared ML platforms and applying machine learning at scale in domains like ads or recommender systems, with a focus on privacy-preserving technologies and agentic AI. Joining this team means contributing to a culture that values reliability, simplicity, and scalability while working alongside world-class engineers dedicated to delivering extraordinary customer experiences.

Skills

Python TensorFlow PyTorch Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL AWS Azure Google Cloud Platform Federated Learning Differential Privacy Transformers LLMs DNNs Model Pruning Compression Quantization Distillation Agentic AI

What you'll do

  • Design and develop model training and fine-tuning infrastructure at scale.
  • Build high-performing, elegant machine learning systems from the ground up.
  • Select appropriate technologies to solve complex challenges efficiently.
  • Define and refine architectures to meet unique ad network challenges.
  • Develop ML products that adhere to Apple's privacy commitments.

What we're looking for

  • Experience building shared ML platforms used by multiple teams.
  • Deep understanding of the ML lifecycle, including training pipelines and deployment patterns.
  • Expertise in deep learning architectures like Transformers and training frameworks such as TensorFlow or PyTorch.
  • Prior experience applying machine learning at scale in advertising or recommender systems.
  • Experience with distributed training and optimization techniques for large-scale models.
  • Ability to build AI/ML tooling for model fine-tuning and infrastructure at scale.
  • Strong technical leadership skills and effective communication with multi-functional teams.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 255 open roles on FindRole.

Listed pay typically runs $171,600–$272,100 across 182 roles with salary data.

Most-posted roles

View all roles at Apple Inc