Machine Learning Engineer, Foundation Model Services

Apple Inc.; Seattle (WA); Hybrid; USD 175,000 - 309,000; Full time

Job summary

Apple Inc. in Seattle is seeking a Software Engineer focused on Machine Learning and AI to build production-grade systems that serve real-time model inferences at scale.

You will collaborate with researchers and product teams to prototype architectures, deploy and operate high-throughput services on cloud platforms using containers and Kubernetes, and write clean, reliable code in Go or Python. Required: 2+ years shipping production software, 5+ years ML/AI experience, and a BS in CS or related

Qualifications

• 2+ years of industry experience shipping production software or ML systems.
• Proficiency in Go or Python and experience with containers and Kubernetes.
• 5+ years in ML technologies (LLMs, NLP, IR, statistics).
• Bachelor’s degree in CS or a related technical field.

Responsibilities

• Build production-grade solutions that launch models serving customers in real time.
• Prototype and develop inference for cutting-edge model architectures with researchers.
• Write high-quality code and operate low-latency services at scale.

Skills

Go

Python

Kubernetes

Docker

Cloud platforms

Communication

ML/LLMs

Education

Bachelor’s degree in Computer Science or related field

Tools

TensorRT-LLM

Job description

Seattle, Washington, United States Machine Learning and AI

Do you think differently? Are you eager to break the status quo, bold and ambitious, unafraid to take risks, and passionate about building best-in-class technology? If so, there's no better place to do it than Apple. The Foundation Model Services team builds the frameworks, services, and tools that run Apple's largest foundation models in production. Our infrastructure powers intelligent experiences across products people use every day — Search, Music, TV, the App Store, Messages, Photos, Spotlight, Safari, Siri, and more — serving millions of queries at incredibly low latency while drawing every ounce of performance from our hardware. Join us and you'll help bring intelligence to billions of users around the world, working on optimizing and serving large language, vision, and speech models at Apple's scale.

Description

Work closely with product teams to build production-grade solutions that launch models serving customers in real time.Partner with foundation model researchers to prototype and develop inference for cutting-edge model architectures, and build tools that help us understand and remove performance bottlenecks across different hardware and use cases.Write high-quality code, learn quickly in a fast-moving field, and grow your impact as you take on larger pieces of the system.

Minimum Qualifications

• 2+ years of industry experience building and shipping production software and/or machine learning systems.
• Proficiency in a modern programming language such as Go or Python.
• Experience deploying and operating services on a cloud platform (AWS, Azure, GCP, or equivalent) using containers and Kubernetes/Docker.
• 5 year+ industry experience in ML technologies (LLMs, Machine Learning, NLP, Information Retrieval, Statistics).
• Experience building or operating high-throughput, low-latency services.
• Strong communication and collaboration skills, with the ability to partner across research and product teams.
• Bachelor’s degree or higher in Computer Science or related technical field.

Preferred Qualifications

• Familiarity with Nvidia TensorRT-LLM, vLLLM, DeepSpeed, Nvidia Triton Server etc.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $175,000 and $308,500, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.