Staff / Sr. Machine Learning Engineer, AI, Search & Knowledge Platforms

Apple Inc.; Cupertino (CA); On-site; USD 185,000 - 325,000; Full time

Benefits offered by this job

Medical and dental coverage

Employee stock programs

Tuition reimbursement

Relocation assistance

Job summary

Apple Inc. is seeking a Staff / Senior Machine Learning Engineer for AI, Search & Knowledge Platforms in the San Francisco Bay Area. You will build scalable extraction and enrichment pipelines that transform web data into structured knowledge powering Siri, Spotlight, Safari and more.

You will apply NLP, entity linking, and ML models to improve data quality and grounding for Apple's foundation models. The role requires cloud-native experience, distributed systems, and hands-on work with

Qualifications

• Bachelor's degree in Computer Science or related technical field.
• 3+ years of software or ML engineering experience.
• Experience with Python, Java, Scala, or Go.
• Background in algorithms, data structures, and distributed systems.
• Experience with AWS and cloud-native environments.
• Experience with large-scale data processing (Spark, Cassandra).
• Experience with containerized environments (Docker, Kubernetes).
• Experience with ML workflows, including feature engineering, training, evaluation and deployment.

Responsibilities

• Build and optimize large-scale extraction and enrichment pipelines powering Apple experiences.
• Design pipelines using LLMs, NLP, and entity linking to deduplicate and contextualize web information.
• Optimize infrastructure for cost, throughput, and reliability including model serving.
• Develop ML classifiers for data quality, deduplication, and content filtering on billions of records.
• Improve extraction quality to support Apple foundation models and reasoning.

Skills

Python

Golang/Java/Scala

Distributed systems

AWS

Spark

Kubernetes

NLP

ML workflows

Education

Bachelor's degree in Computer Science or related field

Tools

Docker

Kubernetes

Spark

Cassandra

TensorRT-LLM

NVIDIA Triton Server

Job description

Staff / Sr. Machine Learning Engineer, AI, Search & Knowledge Platforms

San Francisco Bay Area, California, United States Machine Learning and AI

Imagine what you could do here. At Apple, great ideas quickly become extraordinary products, services, and customer experiences. Bring passion and dedication to your work, and there’s no telling what you could accomplish.Do you want to make Siri and Apple products smarter for our users? The Information Intelligence teams are building groundbreaking technology for algorithmic search, machine learning, natural language processing, and artificial intelligence. The features we build are redefining how hundreds of millions of people use their computers and mobile devices to search and find what they are looking for.Within this organization, our group focuses on web-scale extraction and enrichment, transforming raw web content into structured, high-quality knowledge that powers Apple’s intelligent experiences. We design scalable extraction algorithms, develop advanced web data deduplication techniques, and apply machine learning to process trillions of records and petabytes of data. We also build and maintain Apple’s Knowledge Graph, integrating diverse data sources into a unified representation of the world knowledge.We’re looking for a Machine Learning Engineer with deep expertise in large-scale data and ML infrastructure. You will build and optimize pipelines that extract, process, and serve data artifacts while advancing the ML frameworks and tooling that underpin Apple’s knowledge and search systems.

Description

Join a dynamic team within Apple's Information Intelligence Infrastructure organization that designs, builds, and operates large-scale systems powering search and AI experiences for billions of users. We develop distributed, data-intensive infrastructure that processes web data at global scale, enabling extraction, enrichment, and knowledge graph construction across diverse content such as HTML, PDF, and other unstructured formats.

Responsibilities

• Build and optimize large-scale extraction and enrichment pipelines that transform raw web data into structured knowledge powering Siri, Spotlight, Safari, and other Apple experiences.
• Design pipelines that leverage large language models, advanced NLP, and entity linking frameworks to identify, deduplicate, and contextualize information from the open web.
• Optimize extraction infrastructure for cost, throughput, and reliability including model serving and batch ML workloads.
• Develop ML classifiers for data quality, deduplication, and content filtering across billions of records.
• Improve extraction quality to produce high-fidelity training corpora for Apple's foundation models, directly improving reasoning and grounding capabilities.

Minimum Qualifications

• Bachelor’s degree or higher in Computer Science or related technical field
• 3+ years of experience in software engineering or ML engineering
• Experience with Golang, Java, Scala, or Python
• Background in computer science: algorithms, data structures, and distributed systems
• Experience working in a cloud-native environment such as AWS
• Experience working with large-scale data processing pipelines (Spark, Cassandra, etc.)
• Experience with micro-service architecture in a containerized environment (Docker, Kubernetes, etc.)
• Experience with machine learning workflows, including feature engineering, training, evaluation, deployment and serving

Preferred Qualifications

• Experience with training and fine-tuning large language models
• Experience with optimizing ML training and serving performance, including GPU tuning, batch size optimization, and multi-node scheduling
• Familiarity with Nvidia TensorRT-LLM, vLLM, Nvidia Triton Server, or similar inference frameworks.
• Experience with NLP, information extraction, or web data systems.
• Excellent interpersonal skills, able to work independently as well as in a team

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 $184,700 and $324,800, 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

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.