AIML - Sr Machine Learning Engineer - Answers, Knowledge & Information

Apple • Santa Clara, CA

71d • $181,100 - $318,400

About The Position

Do you want to make Siri and Apple products smarter for our users? The Answers, Knowledge & Information team is redefining how hundreds of millions of people use their devices to get information. We are an Applied ML team pushing the limits of apple intelligence, assistant response ranking, and search technologies, while also responsible for a production service. We are part of a wider effort to power information across a variety of Apple products - including Siri, Spotlight, Safari, Messages, Lookup, and more. In our team, you will be leveraging and improving upon the latest LLM/ML techniques in order to understand queries, rank user intents, rank documents, and generate answers to users' questions. Our team is responsible for training and deploying these models at scale, using the latest advances for online inference optimization.

Requirements

• 8+ years of industry related experience, working in collaborative environments
• Utilizing PyTorch, TensorFlow, and JAX for training and deploying deep learning models
• Understanding product requirements and then translating them into modeling tasks and engineering tasks
• Building models for search relevance ranking, query understanding, and summarization
• BS or MS in Computer Science or related field

Nice To Haves

• PhD in Computer Science, Artificial Intelligence, Machine Learning, Information Retrieval, Data Science or related field
• Experience in RAG, LLM Reasoning, and LLM Agent
• Experience working in a complex organization with multiple collaborators, strong track record in scaling and launching projects in this setting

Responsibilities

• Build world class large language model at scale to power question and answering system for Siri
• Craft and develop core software infrastructure to support natural language understanding system in production
• Design and implement the low latency and high reliability runtime tech stacks for global search
• Design and develop data pipeline and infrastructure for large scale of data processing
• Evaluate and benchmark Apple natural language processing models, and deploy them to serve billions of Apple users

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• A range of discounted products and free services
• Reimbursement for certain educational expenses — including tuition
• Discretionary bonuses or commission payments
• Relocation assistance

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Master's degree