AIML - Sr Machine Learning Engineer, Data and ML Innovation jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 1 day ago

AIML - Sr Machine Learning Engineer, Data and ML Innovation

Apple is seeking a Senior Machine Learning Engineer to contribute to the development of Foundation Models and AI capabilities. The role involves driving ML innovations, designing training pipelines, and collaborating with various teams to enhance model quality and performance.

Artificial Intelligence (AI)Digital MediaElectronicsMedia and EntertainmentAppsBroadcastingDigital EntertainmentFoundational AIMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Model Training & Optimization
Design and implement large-scale LLM pretraining and post-training pipelines, including supervised fine-tuning, preference optimization, and continual learning
Drive model hillclimbing through disciplined experimentation: dataset curation, hyperparameter tuning, and ablation studies
Work on scalable training workflows using distributed frameworks
Evaluation, Reward, and Data Systems
Develop evaluation frameworks for both offline benchmarks and online metrics, covering reasoning, tool use, and task success
Design and maintain verifiers / rubric-based reward systems for agentic tasks and model alignment
Build data pipelines for data generation, filtering, labeling, and replay buffers
Agent & Environment Infrastructure
Build and maintain agent training environments, including tool APIs, simulators, and sandboxed runtimes
Implement environment abstractions to support reinforcement learning and agent evaluation at scale
Collaborate on large scale RL-infra: RL-trainer, rollout system, and containerized environments

Qualification

Large Language ModelsML Engineering ExperienceTraining PipelinesReinforcement LearningSoftware Engineering FundamentalsCreative ThinkingCritical ThinkingCollaboration Skills

Required

5+ years of hands on ML engineering experiences, with at least 1+ years working directly on large language models or generative AI
Bachelor's, Master's, or PhD in Computer Science, Machine Learning, or a related technical field — or equivalent practical experience
Hands-on experience with LLM training workflows, including one or more of: Pretraining or continued pretraining, Supervised fine-tuning (SFT), Preference optimization (e.g., RLHF, DPO, PPO)
Strong software engineering fundamentals: debugging, testing, code reviews, and production reliability
Demonstrated publication records in relevant conferences (e.g., NeurIPS, ICML, ICLR, etc.)

Preferred

Direct experience with agentic systems, including tool use, environment design, or reinforcement learning
Experience with building or operating training environments or simulators (gym-style, tool-based, or sandboxed environments)
Experience with model hillclimbing workflows: systematic experimentation, ablations, dataset iteration, and continuous quality improvement
Ability to work across research and engineering boundaries, turning ideas into scalable systems
Have demonstrated creative and critical thinking with an innate drive to improve how things work. Have a high tolerance for ambiguity

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