CareerPlanSign in

AIML - Staff Machine Learning Engineer, Answers, Knowledge & Information (AKI)

Santa Clara, United States of America💼 Full-time🗓 2025-08-27 → 2026-09-28

Core

Developing LLMs and NLP models to enhance Siri's ability to answer personal domain questions using user documents.

Role type

Staff Machine Learning Engineer (NLP/LLM)

Builds

Personal Q&A models integrated with Siri capabilities

Domain

Consumer technology, Search, Information Retrieval

Deliverable

production ML models

Required skills

Machine Learning, NLP, LLMs, Python, Go, C/C++, ML frameworks (PyTorch, JAX, TensorFlow, XGBoost), Data pipelines, RLHF, Query understanding, Ranking

Preferred skills

Deep learning, Search relevance and ranking, Q&A, Personalization, User behavior modeling, Data-driven decision-making

Technologies

PyTorch, JAX, TensorFlow, XGBoost

Responsibilities

Research, design, implementation, and evaluation of models for Personal Q&A; Implement and extend LLM models with fine-tuning and RLHF; Develop data pipelines; Establish quality metrics and evaluation sets; Collaborate on product requirements; Define long-term technical vision.

Seniority

Staff, hands-on IC with strategic vision

Sourced via apple · Listed on CareerPlan, which tracks 885,000+ jobs from 20+ sources.