AIML - Sr Machine Learning Engineer, Data and ML Innovation
Core
Drive ML innovations for Apple Foundation Models, focusing on large language model training, evaluation, and agentic system development.
Role type
Senior IC machine learning engineer (LLM & agentic systems)
Builds
Large-scale LLM pretraining/post-training pipelines, evaluation frameworks, reward systems, and agent training environments
Domain
Generative AI / Foundation Models / Agentic Systems
Deliverable
production ML models
Required skills
LLM training workflows (pretraining, SFT, RLHF/DPO/PPO), distributed training frameworks, dataset curation, hyperparameter tuning, evaluation framework design, agent environment design, reinforcement learning infrastructure, software engineering fundamentals, publication record in top ML conferences
Preferred skills
Agentic systems experience (tool use, environment design), building/operating training simulators, model hillclimbing workflows, cross-functional research-to-engineering translation
Technologies
Distributed training frameworks, RL infrastructure, containerized environments, tool APIs, sandboxed runtimes
Responsibilities
Design and implement large-scale LLM pretraining and post-training pipelines; Drive model hillclimbing through disciplined experimentation; Develop evaluation frameworks for offline benchmarks and online metrics; Build data pipelines for data generation, filtering, and labeling; Build and maintain agent training environments including tool APIs and simulators; Collaborate on large scale RL infrastructure
Seniority
Senior, hands-on IC