ML Engineer – LLMs, NLP & Generative AI for Search

Apple Inc.; Cupertino (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Apple Inc. is seeking a machine learning engineer/researcher to advance large language models and generative AI across Apple’s AI-powered products.

You will work on open-domain QA, summarization, and sophisticated retrieval/ranking to deliver direct answers aligned with user intent. You will develop domain-specific LLMs, fine-tune and evaluate models, and translate product needs into scalable ML solutions, collaborating with researchers and data scientists to push state-of-the-art into

Qualifications

• Experience modeling user behavior including personalization and online learning.
• Experience with NLP tasks for RAG, prompt engineering, data collection and model evaluation.
• Proficiency in Python and at least one of the deep learning frameworks such as TensorFlow, PyTorch, or JAX.

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

Skills

Personalization

NLP tasks

Prompt engineering

Data collection

Fine-tuning

Model evaluation

Python

Education

PhD in Computer Science, AI, ML

Tools

TensorFlow

PyTorch

JAX

Job description

Apple Inc. is seeking a machine learning engineer/researcher to advance large language models and generative AI across Apple’s AI-powered products.

You will work on open-domain QA, summarization, and sophisticated retrieval/ranking to deliver direct answers aligned with user intent. You will develop domain-specific LLMs, fine-tune and evaluate models, and translate product needs into scalable ML solutions, collaborating with researchers and data scientists to push state-of-the-art into