AIML Researcher/Engineer - Foundation Model Post-Training
Core
Design and iterate on end-to-end post-training strategies (including Reinforcement Learning) to transform large language models into intelligent assistants for Apple products.
Role type
Senior IC machine-learning researcher (foundation model post-training)
Builds
Large-scale frontier foundation models powering Apple ecosystem products
Domain
Deep learning, Large Language Models (LLMs), Reinforcement Learning
Deliverable
production ML models
Required skills
Deep learning expertise, LLM post-training, Reinforcement Learning, Python programming, JAX or PyTorch, distributed training, transformers architectures, automated data filtering, curriculum learning, evaluation methodologies
Preferred skills
Training state-of-the-art large models at scale, improving model performance on complex reasoning tasks (math, coding, logic), experience with various transformers architectures and their transformations