AIML - Staff Machine Learning Engineer, Siri and Information Intelligence

Apple; Santa Clara (CA); On-site; USD 207,000 - 379,000; Full time

Job summary

Join a forward-thinking company at the forefront of artificial intelligence and machine learning! As part of a dynamic team, you will engage in groundbreaking research and development of state-of-the-art technologies that redefine user interactions with devices. This role offers the unique opportunity to influence product development through innovative machine learning solutions and large language models. Collaborate with top researchers and engineers to create cutting-edge applications that enhance user experiences and drive the future of technology. If you're passionate about AI and eager to make a significant impact, this is the perfect opportunity for you.

Benefits

Comprehensive medical and dental coverage

Retirement benefits

Discounted products and free services

Educational reimbursement

Employee stock purchase plan

Qualifications

• 10+ years of experience in machine learning or deep learning.
• Hands-on experience with deep learning and LLM model development.

Responsibilities

• Conduct research on deep learning and LLMs for AI-powered products.
• Develop and evaluate domain-specific Large Language Models.

Skills

Machine Learning

Deep Learning

Natural Language Processing (NLP)

Python

TensorFlow

PyTorch

JAX

Model Fine-tuning

Prompt Engineering

Education

Master’s in Computer Science

PhD in Computer Science

Tools

Deep Learning Frameworks

Job description

Summary: The AIML Information Intelligence team is creating groundbreaking technology for artificial intelligence, machine learning, and natural language processing! The features we create are redefining how hundreds of millions of people use their computers and mobile devices to search and find what they are looking for. Our universal search engine powers search features across a variety of Apple products, including Siri, Spotlight, Safari, Messages, and Lookup. We also develop state-of-the-art generative AI technologies based on Large Language Models to power innovative features in both Apple’s devices and services on the cloud. 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 the 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. This role will have the following responsibilities:

1. 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.
2. 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.
3. Conducting applied research to transfer the cutting edge research in generative AI to production-ready technologies.
4. Understanding product requirements, translating them into modeling tasks and engineering tasks.
5. Stay up to date with the latest advancements and research in deep learning and large language models.

Minimum Qualifications

• Master’s in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 10 years of work experience in machine learning, deep learning, or related field.
• 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.

Key Qualifications

Preferred Qualifications

• PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• At least 2 years 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 least 10 years of experience leading complex cross-functional research projects and influencing research direction.
• At least 10 years of experience with large-scale model training, optimization, and deployment.
• One or more scientific publications in various conferences and journals.
• Consistent track record of researching, inventing, and/or shipping advanced machine learning models.
• Outstanding communication and interpersonal skills with the ability to work with cross-functional teams.

Education & Experience

Additional Requirements

Pay & Benefits

• 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 $207,800 and $378,700, 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 take affirmative action to ensure 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.