AIML - Machine Learning Researcher, Post-Training for Foundation Models
Core
Researching and developing post-training algorithms (SFT, RL) to transform pre-trained foundation models into helpful, high-quality models optimized for Apple silicon and private, personal experiences.
Role type
Senior IC machine learning researcher (foundation model post-training)
Builds
Frontier foundation models powering intelligent experiences on Apple devices
Domain
Artificial Intelligence / Large Language Models / Apple Silicon
Deliverable
production ML models
Required skills
deep learning, LLMs, supervised fine-tuning (SFT), reinforcement learning, Python, JAX/PyTorch
Preferred skills
post-training algorithms, synthetic data generation, reasoning tasks, distributed training, evaluation methodologies
Technologies
JAX, PyTorch
Responsibilities
Design end-to-end post-training recipes combining SFT and RL; Develop novel algorithms for preference optimization and safety; Research data generation and filtering methods; Design robust evaluation frameworks for model utility; Collaborate with pre-training and product teams
Seniority
Senior, hands-on IC
