ASE - Machine Learning Engineer, MLPT

Apple Inc.; Santa Clara, Northern (CA, KY); Hybrid; USD 185,000 - 325,000; Full time

Benefits offered by this job

Medical & dental coverage

Employee stock programs

Tuition reimbursement

Job summary

Apple Inc. in Santa Clara, CA is seeking a Senior Machine Learning Engineer to advance Open Domain Question Answering, summarization, and knowledge-driven AI across Apple’s platforms.

You will build and refine large language models, optimize retrieval and ranking, and push state-of-the-art research into scalable production. The role focuses on fine-tuning, evaluation, and integration of domain-specific models for NLP tasks, collaborating with researchers and engineers to deliver impactful AI

Qualifications

• 2+ years of deep learning or LLM development for NLP tasks and RAG apps.
• Experience with Python and at least one DL framework (TF, PyTorch, or JAX).
• Master’s in Computer Science, AI, ML, or related field.

Responsibilities

• Research and development on deep learning and LLMs for Apple products
• Develop, fine-tune, and evaluate domain-specific LLMs for NLP tasks
• Transfer generative AI research to production-ready tech
• Translate product requirements into modeling and engineering tasks
• Stay current with advances in DL and LLMs

Skills

Deep learning

LLM development

Python

Prompt engineering

Large-scale training

Education

Master's degree in CS/AI/ML

PhD in related field (preferred)

Tools

TensorFlow

PyTorch

JAX

Job description

Senior Machine Learning Engineer, Apple Search & Knowledge Platforms

Santa Clara, California, United States Machine Learning and AI

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 be doing large scale machine learning and deep learning research and development to improve Open Domain Question Answering (using both structured knowledge graph data and unstructured web data) and Summarization as well as developing fundamental building blocks needed for Artificial Intelligence. This involves developing sophisticated machine learning and large language models (LLMs) to understand user queries, retrieve and rank relevant documents across multiple sources and synthesize information across documents to provide user with a direct answer that best satisfies their intent and information seeking needs. Additionally, you will research and develop the state-of-the-art LLMs for summarizing personal data such as emails, messages, and notifications.You will also work with researchers and data scientists to develop, fine-tune, and evaluate domain specific Large Language Models for various tasks and applications in Apple’s AI powered products and conduct applied research to transfer the cutting edge research in generative AI to production ready technologies.

Description

As a member of our fast-paced group, you'll have the unique and rewarding opportunity to shape upcoming products from Apple. We are looking for highly motivated machine learning engineers and researchers having strong machine learning and deep learning fundamentals with hands-on experience in fine-tuning deep learning and large language models.

Responsibilities

• Conduct research and development on state-of-the-art deep learning and large language models for various tasks and applications in Apple’s AI-powered products
• Developing, fine-tuning, and evaluating domain-specific Large Language Models for various NLP tasks including summarization, question answering, search relevance/ranking, entity linking and query understanding problems
• Conducting applied research to transfer the cutting edge research in generative AI to production ready technologies
• Understanding product requirements, translate them into modeling tasks and engineering tasks
• Stay up to date with the latest advancements and research in deep learning and large language models

Minimum Qualifications

• 2+ years of experience working with Deep learning or LLM model development for various NLP tasks and RAG applications including prompt engineering, training data collection and generation, model fine-tuning and model evaluation.
• Experience working with Python and at least one of the deep learning frameworks such as TensorFlow, PyTorch, or JAX.
• Master’s in Computer Science, Artificial Intelligence, Machine Learning, or a related field.

Preferred Qualifications

• PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 4+ years of experience with large-scale model training, optimization, and deployment
• One or more scientific publications in various conferences and journals
• Outstanding communication and interpersonal skills with ability to work with cross-functional teams.
• 1+ year of experience in various state-of-the-art techniques related to LLM fine-tuning in 1 or more of the following areas:
• Supervised Fine-tuning (SFT) with Rejection Sampling
• Preference-based fine-tuning techniques (e.g RLHF, Reward model, DPO, PPO, GRPO etc.)
• Parameter efficient fine-tuning techniques (e.g LoRA)
• Hallucination reduction and factual accuracy improvements
• Designing and implementing safety guardrails

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 $184,700 and $324,800, 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.