Staff / Sr. Machine Learning Engineer, AI, Search & Knowledge Platforms
Core
Build and optimize large-scale extraction and enrichment pipelines to transform raw web data into structured knowledge powering Siri, Spotlight, Safari, and other Apple experiences.
Role type
Staff/Senior Machine Learning Engineer (Search & Knowledge Platforms)
Builds
Scalable extraction algorithms, web data deduplication techniques, and Apple's Knowledge Graph
Domain
Consumer Technology / Web-scale Data & AI
Deliverable
production ML models
Required skills
Large-scale data processing, ML infrastructure optimization, LLM fine-tuning, NLP, distributed systems, cloud-native environments, model training and evaluation
Preferred skills
GPU tuning, multi-node scheduling, inference frameworks (TensorRT-LLM, vLLM, Triton), information extraction
Technologies
Golang, Java, Scala, Python, AWS, Spark, Nvidia TensorRT-LLM, vLLM, Nvidia Triton Server
Responsibilities
Design pipelines leveraging LLMs and NLP to identify, deduplicate, and contextualize information from the open web; Develop ML classifiers for data quality, deduplication, and content filtering; Improve extraction quality to produce high-fidelity training corpora for foundation models
Seniority
Staff/Senior, hands-on IC
