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 theres no telling what you could 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 this organization our group focuses on web-scale extraction and enrichment transforming raw web content into structured high-quality knowledge that powers Apples 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 Apples Knowledge Graph integrating diverse data sources into a unified representation of the world 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 Apples knowledge and search systems.

Join a dynamic team within Apples 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.

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

Bachelors degree or higher in Computer Science or related technical fieldn3 years of experience in software engineering or ML engineeringnExperience with Golang Java Scala or PythonnBackground in computer science: algorithms data structures and distributed systemsnExperience working in a cloud-native environment such as AWSnExperience working with large-scale data processing pipelines (Spark Cassandra etc.)nExperience with micro-service architecture in a containerized environment (Docker Kubernetes etc.)nExperience with machine learning workflows including feature engineering training evaluation deployment and serving

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