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

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.

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