Apple · Santa Clara

AIML - Staff Machine Learning Engineer - Reinforcement Learning

6/23/2025

Description

In this role, you will work on LLM based question answering and Apple Intelligence features to provide concise, accurate, and grounded information to users to help them complete their tasks quickly on Apple devices. Your core responsibilities will include: * Designing and developing advanced Reinforcement Learning technologies in the post-training of generative model, and delivering the end-user experience. * Driving cross-functional technical initiatives, collaborating with research, engineering and production teams to translate theoretical advances into deployable systems. * Developing novel and cutting-edge RL algorithms and improving existing ones. * Staying up to date with the latest RL research and integrate best practices into the team's workflow. * Working on the end-to-end ML lifecycle: algorithm design and implementation, data collection, model training, evaluation, and deployment.

Qualifications

  • 10+ years of ML experiences in search, natural language processing/understanding. Conversational AI.
  • Proven experience for LLM post training, including but not limited to SFT, RLHF, RLAIF, Reward Modeling, Chain-of-thought, agentic LLM.
  • Hands-on experience building RL pipelines and training agents in simulation or real-world environments.
  • Growth mindset and ability to learn new technologies
  • MS or Ph.D. in Computer Science, Machine Learning with a specialty in reinforcement learning, or a related field

Preferred Qualifications

  • Deep expertise in reinforcement learning-based post-training on LLM models, reward modeling, RLHF, RLAIF, Chain-of-thought, and agentic AI R&D.
  • Deep understanding of cutting edge RL algorithms and large language model.
  • Deep understanding in LLM pre-training, post-training.
  • Strong product intuition and ownership
  • Excellent communication skills

Benefits

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $212,000 and $386,300, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Application

View listing at origin and apply!