ASE - Machine Learning Engineer, MLPT

Cupertino, California, United States

The Information Intelligence team is creating groundbreaking technology for artificial intelligence, machine learning and natural language processing. The features we create redefine 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 Apple products including Siri, Spotlight, Safari, Messages and Lookup. We develop state-of-the-art generative AI technologies based on Large Language Models to power features in both Apple’s devices and services on the cloud.

As part of this group, you will conduct large-scale machine learning and deep learning research and development to improve Open Domain Question Answering (using structured knowledge graph data and unstructured web data) and Summarization, and develop fundamental building blocks for Artificial Intelligence. This involves developing machine learning and large language models (LLMs) to understand user queries, retrieve and rank relevant documents across sources, and synthesize information across documents to provide a direct answer that best satisfies user intent.

You will also research and develop state-of-the-art LLMs for summarizing personal data such as emails, messages, and notifications. You will 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 cutting-edge generative AI research to production-ready technologies.

• 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.
• Develop, fine-tune, and evaluate domain-specific Large Language Models for NLP tasks including summarization, question answering, search relevance/ranking, entity linking and query understanding.
• Conduct applied research to transfer cutting-edge generative AI to production-ready technologies.
• Understand 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.

• Master’s in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Experience with deep learning or LLM model development for NLP tasks and RAG applications including prompt engineering, training data collection and generation, model fine-tuning and evaluation.
• Experience with Python and at least one deep learning framework such as Tensor Flow, PyTorch, or JAX.

• PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• At least 1 year of experience in state-of-the-art LLM fine-tuning techniques (e.g., SFT with Rejection Sampling, RLHF, Reward models, DPO, PPO, GRPO, etc.).
• At least 4 years of experience with large-scale model training, optimization, and deployment.
• Experience with parameter-efficient fine-tuning techniques (e.g., LoRA).
• Experience with safety guardrails, hallucination reduction and factual accuracy improvements.
• One or more scientific publications in conferences and journals.
• Outstanding communication and interpersonal skills with ability to work with cross-functional teams.

At Apple, base pay is one part of our total compensation package and is determined within a range. The base pay range for this role is between $181,100 and $272,100, 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. This includes discretionary restricted stock units and the ability to purchase Apple stock through the Employee Stock Purchase Plan. Benefits include comprehensive medical and dental coverage, retirement benefits, access to discounted products and free services, and reimbursement for certain educational expenses related to advancing your career  role may be eligible for discretionary bonuses or commissions and relocation benefits.

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 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.

Apple accepts applications to this posting on an ongoing basis.