Staff/Senior Machine Learning Engineer, Search & Knowledge Platforms

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

Job summary

Apple's AI, Search & Knowledge Platforms team builds world‑class products powering Siri, Apple Intelligence, Safari and iMessage. We are looking for a versatile Machine Learning and Software Engineer to improve the search and ranking stack with ML systems, robust heuristics, and LLMs.

You will tackle end‑to‑end challenges—from data pipelines and model training to prompt engineering and evaluation—while collaborating across teams to ship reliable, high‑quality knowledge answers that meet user

Qualifications

• Bachelor’s in Computer Science, AI, ML, or related field.
• 6+ years shipping Search and ML systems.
• Strong programming skills in Python, Go or similar languages.
• Experience building scalable data pipelines and backend systems.
• Proficiency in data science methodologies, including A/B testing and evaluation metrics for ML systems.
• Experience with prompt engineering.
• Strong product sense translating ambiguous product requirements into concrete tasks.
• Excellent communication with cross-functional partners.

Responsibilities

• Identify opportunities to improve search and ranking stack.
• Develop ML systems, heuristics, and LLMs to understand queries.
• Rank relevant documents and provide direct answers.
• Collaborate across teams to ship features and improvements.
• Evaluate models with offline and online metrics.

Skills

Python

Go

Data pipelines

ML systems

A/B testing

Prompt engineering

Product sense

Communication

Algorithms

Software design

Education

Bachelor's in CS/AI/ML

Master's/PhD in CS/AI/ML

Tools

RAG architectures

LLM deployment tooling

NLP libraries

Job description

The AI, Search & Knowledge Platforms team builds amazing products and services for Apple's customers while serving as a foundational partner to teams across Apple. The team delivers world-class AI, search, and knowledge systems powering Siri, Apple Intelligence, Safari, and iMessage, and operates the foundational platforms and infrastructure that keep these intelligent experiences running at hyperscale. As part of this group, you will focus on identifying and addressing opportunities to improve the quality and performance of the search and ranking stack that powers world knowledge question answering. This involves developing sophisticated machine learning systems, robust software heuristics, and large language models (LLMs) to understand user queries, retrieve and rank relevant documents, and provide users with direct answers that best satisfy their intent. You will work across the entire stack, applying a broad skill set that ranges from data science and opportunity analysis to ML system design, pipeline development, and quality evaluation.

Description

As a member of our fast-paced group, you will have the unique and rewarding opportunity to shape upcoming products from Apple. We are looking for a highly motivated and versatile Machine Learning and Software Engineer with a broad skill set. The ideal candidate combines strong engineering fundamentals with a keen product sense, and is a quick learner who is comfortable jumping into entirely new codebases and architectures. You must be an effective communicator and an adaptable problem solver, willing to take on whatever represents the optimal solution for the product, whether that means training machine learning models, writing robust software heuristics, or iterating on prompt engineering.

Minimum Qualifications

• Bachelor’s in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 6+ years experience in shipping Search and ML systems
• Strong programming skills in Python, Go or similar languages, with a solid foundation in software design, data structures, and algorithms.
• Experience building scalable data pipelines and backend systems, coupled with the ability to quickly navigate and contribute to large, complex, and unfamiliar codebases.
• Proficiency in data science methodologies, including A/B testing, statistical hypothesis testing and defining offline and online evaluation metrics for ML systems.
• Experience with prompt engineering
• Strong product sense with the ability to translate ambiguous product requirements into concrete engineering and modeling tasks.
• Excellent communication skills, with the ability to back up decisions with rigorous data analysis and present findings clearly to cross-functional partners.

Preferred Qualifications

• Master’s/Ph.D in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 4+ years experience in shipping Search and ML systems
• Hands-on experience in Information Retrieval, Search Ranking, or Natural Language Processing (NLP).
• Demonstrated experience designing, building, or working with Retrieval-Augmented Generation (RAG) architectures and applications.
• Familiarity with techniques for optimizing the inference performance, latency, and resource efficiency of LLMs and large machine learning systems in production environments (e.g., model quantization, caching, distributed inference).
• Experience with training and deploying machine learning models