Senior Machine Learning Engineer, Proactive

Apple • Santa Clara, CA

13d

About The Position

At Apple, machine learning powers experiences that anticipate what people need before they ask. We're looking for a Machine Learning Engineer to help build the next generation of intelligent search and AI experiences technology that understands user intent, context, and personal information while preserving privacy. In this role, you'll design, train, optimize, and deploy large language models, semantic retrieval systems, and ranking models that power relevant, personalized, context-aware search across Apple's ecosystem. You'll work at the intersection of search, retrieval, natural language processing, on-device AI, and generative AI to shape the future of intelligent assistants and proactive experiences. You'll design, train, fine-tune, and optimize transformer-based language models for on-device deployment, and build semantic retrieval, embedding, reranking, and retrieval-augmented generation systems that improve search quality and AI-powered experiences. You'll develop models for query understanding, intent prediction, personalization, and ranking, while researching new approaches to model compression, quantization, and low-latency inference. You'll partner with engineers, researchers, product managers, and designers to bring new AI capabilities from research into production driving technical strategy and leading projects from early exploration through large-scale deployment. This is an opportunity to explore new applications of foundation models, multimodal AI, and agentic retrieval, shaping the next generation of proactive, intelligent user experiences.

Requirements

• Bachelor's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
• 5+ years of industry or research experience developing machine learning systems.
• Background in machine learning, deep learning, natural language processing, information retrieval, search, recommender systems, or generative AI.
• Experience training, fine-tuning, or deploying transformer-based models and large language models.
• Programming skills in Python and/or C/C++.
• Experience building production-quality software using modern machine learning frameworks such as PyTorch, JAX, or TensorFlow.

Nice To Haves

• Master's or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
• Experience optimizing machine learning models for resource-constrained environments, including model compression, quantization, pruning, and knowledge distillation.
• Experience with on-device machine learning or mobile inference frameworks.
• Experience building retrieval-augmented generation, vector search, embedding retrieval, or semantic search systems.
• Experience working with transformer architectures such as BERT, T5, Llama, Gemma, Mistral, or other foundation models.
• Experience evaluating language models, designing AI quality metrics, and building offline evaluation pipelines.
• Experience building large-scale production search, recommendation, or personalization systems.
• Ability to prototype ideas, solve ambiguous problems, and deliver production-quality machine learning solutions.

Responsibilities

• Design, train, fine-tune, and optimize transformer-based language models for on-device deployment.
• Build semantic retrieval, embedding, reranking, and retrieval-augmented generation systems that improve search quality and AI-powered experiences.
• Develop models for query understanding, intent prediction, personalization, and ranking.
• Research new approaches to model compression, quantization, and low-latency inference.
• Partner with engineers, researchers, product managers, and designers to bring new AI capabilities from research into production.
• Drive technical strategy and lead projects from early exploration through large-scale deployment.
• Explore new applications of foundation models, multimodal AI, and agentic retrieval.

What This Job Offers

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