Senior Machine Learning Engineer, Proactive

Socket.dev; Santa Clara (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Apple is seeking a Machine Learning Engineer to design, train, and deploy transformer-based language models for on-device deployment, enabling private, context-aware search across Apple’s ecosystem.

You will build semantic retrieval, embedding, and retrieval-augmented generation systems, while researching model compression and low-latency inference. Collaborate with cross-functional teams to bring AI capabilities into production.

Qualifications

• Bachelor's degree in CS/ML/AI or related field.
• 5+ years of industry or research experience in ML systems.
• Experience training/finetuning/deploying transformer models and LLMs.
• Proficiency in Python and/or C/C++ with modern ML frameworks.

Responsibilities

• Design, train, and optimize transformer-based language models for on-device deployment.
• Build semantic retrieval, embedding, reranking, and retrieval-augmented generation systems.
• Research model compression, quantization, and low-latency inference techniques.
• Collaborate with engineers, researchers, product managers, and designers to move AI capabilities into production.

Skills

Python

C/C++

NLP

On-device AI

IR concepts

Education

Bachelor's degree

Master's degree

Tools

PyTorch

JAX

TensorFlow

Job description

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..<\/p>

DESCRIPTION

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.<\/p>

MINIMUM QUALIFICATIONS

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++, with experience building production-quality software using modern machine learning frameworks such as PyTorch, JAX, or TensorFlow.<\/p>

PREFERRED QUALIFICATIONS

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.<\/p>