SIGN IN
AIML - Machine Learning Researcher, Post-training for Foundation Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 10 hours ago

AIML - Machine Learning Researcher, Post-training for Foundation Models

Apple is a company responsible for building foundation models, and they are seeking a Machine Learning Researcher for their Post-training team. The role focuses on transforming pre-trained models into high-quality models, involving research into training recipes and evaluation methodologies.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Recipe Development: Design and iterate on end-to-end post-training recipes, combining SFT, Reinforcement Learning and reasoning regimes to achieve specific model behaviors and capabilities
Algorithm Research: Develop and implement novel algorithms for preference optimization, model steering, and safety
Data Strategy: Research methods for high-quality human and synthetic data generation, automated data filtering, and curriculum learning to improve instruction following and reasoning capabilities
Evaluation: Design robust evaluation frameworks to measure model helpfulness, factuality, and utility, moving beyond static benchmarks to capture real-world performance
Collaboration: Work closely with pre-training teams to inform architecture choices and with product teams to understand user requirements

Qualification

Deep LearningPost-trainingReinforcement LearningPythonJAXPyTorchData Quality AssessmentEvaluation MethodologiesReasoning ResearchDistributed TrainingCommunicationCollaboration

Required

Demonstrated expertise in deep learning with a focus on LLMs, post-training, or reinforcement learning, backed by a strong publication record or real world experiences and accomplishments in these or closely related domains
Proficient programming skills in Python and one of the deep learning frameworks such as JAX or PyTorch
PhD or equivalent practical experience, in Computer Science, Machine Learning, or a related technical field

Preferred

Proven track record in post-training: Specialization in post-training algorithms, techniques, and best practices for large foundation models with proven track record
Post-training data: Deep experiences with human data labeling, synthetic data generation and data quality assessment for foundation models
Evaluation methodologies: Deep experience in evaluating data and training recipe and deeply understand the model building iterative process and life cycle
Reasoning Research: Experience in improving model performance on reasoning tasks (math, coding, logic)
Scale & Systems: Experience training SOTA large models at scale and familiarity with distributed training challenges, and understand the trade-offs
Strong communication and collaborative skills: Strong communication skills and a passion for collaboration within and across teams

Benefits

Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation

Company

Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.

H1B Sponsorship

Apple has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)

Funding

Current Stage
Public Company
Total Funding
$5.67B
Key Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity

Leadership Team

leader-logo
Tim Cook
CEO
leader-logo
Craig Federighi
SVP, Software Engineering
Company data provided by crunchbase