Apple · 5 hours ago
AIML - Machine Learning Engineer, Data and ML Innovation
Cupertino, CA
Full-time
Onsite
Mid, Senior Level
$181K/yr - $318K/yr
3+ years expApple is part of the AI and Machine Learning organization, creating groundbreaking technology for large-scale ML systems. They are seeking a Machine Learning Engineer to develop advanced LLM-based agents and collaborate with cross-functional teams to push the frontier of agent modeling and reinforcement learning.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Develop and improve LLM-based agent models that can plan, reason, and act across complex, multi-turn tasks
Design mid-training pipelines to enhance agentic capabilities such as tool-use, reflection, and long-horizon reasoning
Explore and implement reinforcement learning and feedback-based fine-tuning to align agent behaviors with human and system objectives
Build simulation and evaluation environments for measuring multi-turn success rate, planning efficiency, and robustness of agent interactions
Develop scalable frameworks for synthetic agentic data generation, including user–assistant–tool trajectories, environment modeling, and self-play
Collaborate with modeling, data, and infrastructure teams to integrate new training signals and improve overall system performance
Stay on top of cutting-edge research in agentic LLMs, reasoning, tool-use, mid-training, and reinforcement learning
Qualification
Required
Strong background in machine learning, natural language processing, or reinforcement learning with a focus on LLM or agentic systems
Solid understanding of LLM training paradigms (pre-training, mid-training, RL, post-training) and multi-modal or tool-use-enhanced architectures
Excellent programming skills and experience with Python and modern deep learning frameworks (PyTorch, JAX, or equivalent)
Ability to design, run, and analyze large-scale training and evaluation experiments
M.S. in Computer Science, Machine Learning, Artificial Intelligence, or related fields
Preferred
3+ years of hands-on experience developing agentic LLMs, tool-use systems, or RL-based fine-tuning pipelines
Experience in multi-turn data generation, simulation environments, or agent evaluation frameworks
Deep familiarity with self-improvement loops, reward modeling, or interactive reinforcement learning
Strong publication record in top-tier venues (e.g., NeurIPS, ICLR, ICML, AAAI, ACL, EMNLP)
Proven ability to work in collaborative, cross-functional ML teams
Ph.D. in Machine Learning, NLP, Reinforcement Learning, or related disciplines
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
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
10001+ employees
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 CompanyTotal Funding
$5.67BKey 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
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase