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.

We are part of the Apple's Information Intelligence Infrastructure organization, which designs, builds, and operates large-scale systems powering search and AI experiences for billions of users. 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 across 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.

Minimum Qualifications

• 4+ years of experience in at least 1 of the following: fine-tuning large language models; building large-scale extraction and enrichment pipelines; or building and optimizing ML infrastructure for training or inference.
• Experience taking machine learning models to production, including model training and evaluation
• 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 with large-scale data processing pipelines (e.g., Spark or similar distributed processing frameworks)
• Bachelor’s degree or higher in Computer Science or related technical field

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