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

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.","responsibilities":"Build and optimize large-scale extraction and enrichment pipelines that transform raw web data into structured knowledge powering Siri, Spotlight, Safari, and other Apple experiences.

Design pipelines that leverage large language models, advanced NLP, and entity linking frameworks to identify, deduplicate, and contextualize information from the open web.

Optimize extraction infrastructure for cost, throughput, and reliability including model serving and batch ML workloads.

Develop ML classifiers for data quality, deduplication, and content filtering across billions of records.

Improve extraction quality to produce high-fidelity training corpora for Apple's foundation models, directly improving reasoning and grounding capabilities.

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

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

Pulled from the full job description Employee stock purchase plan Health insurance Retirement plan Dental insurance RSU, At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $175,000 and $308,500, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits