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

Apple • Seattle, WA

3d

About The Position

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. We are 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.

Requirements

• 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

Nice To Haves

• 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

Responsibilities

• Build and optimize pipelines that extract, process, and serve data artifacts.
• Advance the ML frameworks and tooling that underpin Apple’s knowledge and search systems.
• Design scalable extraction algorithms.
• Develop advanced web data deduplication techniques.
• Apply machine learning to process trillions of records and petabytes of data.
• Build and maintain Apple’s Knowledge Graph, integrating diverse data sources into a unified representation of the world knowledge.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior