A

Apple

Designs phones, computers and their operating systems

AI/ML Distinguished Engineer - Foundation Model

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeCupertino, CA, USANo salary listedIn-person

About the Job

Requirements

  • An MS or PhD in Computer Science, Machine Learning, or a related technical field, or equivalent industry experience.
  • 15 or more years of experience building large-scale machine learning or distributed systems, with a track record of technical leadership and industry-wide or company-wide impact.
  • Hands-on expertise in foundation model inference engines, with a record of improving performance, efficiency, and reliability at scale.
  • Experience supporting diverse foundation model inference use cases with different throughput, latency, cost, and quality constraints.
  • Breadth beyond inference, including the ability to contribute in adjacent systems areas such as training infrastructure, data systems, or evaluation.
  • Deep understanding of GPU, TPU, and accelerator architecture, distributed systems, and model optimization, including quantization, distillation, compilation, and serving.
  • Proficiency with machine learning frameworks such as JAX and PyTorch, and with inference and serving stacks.
  • Experience leading diverse engineers in setting vision and driving execution, including prioritization for milestone deliveries.
  • Experience mentoring junior and senior engineers.
  • Experience partnering with machine learning researchers and modeling teams to productionize research.

Responsibilities

  • Set and drive the technical vision and roadmap for the foundation model team's inference engine and the systems around it, used for training, evaluation, and LLM-as-judge workloads.
  • Lead deep work on inference performance, efficiency, and reliability, including throughput and latency optimization, batching and scheduling, quantization, speculative decoding, KV-cache management, memory and compute efficiency, and hardware-aware optimization.
  • Architect inference systems that support internal use cases including data generation and rollouts for training, offline and large-scale evaluation, and judge and reward scoring across text, image, speech, and multimodal models with distinct throughput, cost, and quality constraints.
  • Extend impact into adjacent systems areas, including training infrastructure, data pipelines, and evaluation harnesses.
  • Partner with teams that depend on the engine, translating their needs into a coherent platform, clear interfaces, and a prioritized roadmap.
  • Work with machine learning researchers and modeling teams to co-design models and systems, and bring state-of-the-art techniques from prototype into the development loop reliably.
  • Lead engineers across teams in setting direction and executing against it; align stakeholders, resolve technical trade-offs, and make decisions that keep large efforts moving.
  • Drive prioritization and milestone delivery across competing demands, balancing near-term research needs against long-term platform investment.
  • Mentor and grow junior and senior engineers, establish engineering standards, review designs, and multiply organizational impact.

Desired Qualifications

  • Experience building or leading inference systems for large language models and multimodal foundation models at scale.
  • Experience with inference in training, evaluation, or reinforcement-learning loops, such as large-scale rollouts, offline evaluation, or LLM-as-judge and reward scoring.
  • Familiarity with Kubernetes, Docker, cloud platforms such as AWS, GCP, and Azure, and distributed computing frameworks.
  • A history of defining technical strategy that shaped an organization's or the industry's direction.

About the Company

Apple

Designs phones, computers and their operating systems

Apple designs and sells consumer electronics, from iPhone smartphones and Mac computers to tablets and wearable devices. Its operating systems manage apps and built-in functions for communication and work, while iCloud stores files and synchronizes data across devices. Customers buy additional digital services through subscriptions such as Apple Music and Apple TV, and developers distribute software through the App Store. Apple earns money from hardware sales, subscriptions and digital transactions, selling directly to customers and through retailers and mobile carriers around the world. It was founded in 1976 and has its headquarters in Cupertino, California.

Company size
10,001+
Company stage
IPO
Headquarters
Cupertino, CA
Founded
1976

Recently Posted Jobs

Explore more opportunities at Apple

See more jobs