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

As a member of our fast-paced group, you’ll have the unique and rewarding opportunity to shape upcoming products from Apple. Our team includes a diversity of backgrounds from applied machine learning engineers with a focus on ML and LLM to experienced distributed systems engineers. As such, we are looking for candidates with applied machine learning experience and strong software engineering skills.

MS in Computer Science or related field
5+ years of industry related experience, working in collaborate 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 machine-learned models for search relevance ranking, query understanding, and summarization

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

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.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $171,600 and $302,200, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.