Senior ML Engineer - On-Device AI & Retrieval

Apple Inc.; Cupertino, Northern (CA, KY); Hybrid; USD 150,000 - 278,000; Full time

Job summary

Apple in Cupertino, California is seeking a Senior Machine Learning Engineer to help build the next generation of intelligent search and AI experiences while preserving user privacy. You will work with large language models, semantic retrieval systems, and ranking models to design, optimize, and deploy personalized experiences across Apple's ecosystem.

Responsibilities include designing, optimizing, and deploying semantic retrieval, embedding, reranking, and retrieval-augmented generation

Qualifications

• Bachelor's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
• Background in machine learning, deep learning, natural language processing, information retrieval, search, recommender systems, or generative AI.
• Experience with semantic retrieval, embedding models, reranking, or retrieval-augmented generation systems.
• Programming skills in Python and/or C/C++, with experience building production-quality software using modern ML frameworks such as PyTorch, JAX, or TensorFlow.

Responsibilities

• Work with semantic retrieval systems and ranking models to design, optimize, and deploy relevant, personalized, and context-aware experiences across Apple's ecosystem.
• Build semantic retrieval, embedding, reranking, and retrieval-augmented generation systems that improve search quality and AI-powered experiences.
• Improve the underlying software infrastructure and optimize solutions for low-latency inference.
• Partner with engineers, researchers, product managers, and designers to bring AI capabilities from research into production, driving technical strategy and exploring new applications of foundation models, multimodal AI, agentic retrieval, and personalized intelligence.

Skills

ML background

NLP & IR knowledge

Python proficiency

C/C++ proficiency

Production software

Education

Bachelor's degree in Computer Science, ML, or AI

Tools

PyTorch

JAX

TensorFlow

Semantic retrieval

Embedding models

Reranking

Retrieval-Augmented Generation

Job description

Apple in Cupertino, California is seeking a Senior Machine Learning Engineer to help build the next generation of intelligent search and AI experiences while preserving user privacy. You will work with large language models, semantic retrieval systems, and ranking models to design, optimize, and deploy personalized experiences across Apple's ecosystem.

Responsibilities include designing, optimizing, and deploying semantic retrieval, embedding, reranking, and retrieval-augmented generation