AIML - Staff Machine Learning Engineer

Santa Clara, California, United States

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, helping them complete tasks quickly on Apple devices. Your core responsibilities will include:

• Designing and developing advanced Reinforcement Learning technologies in the post-training of generative models, 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 integrating 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.

Minimum Qualifications

• 10+ years of ML experience in search, natural language processing/understanding, and conversational AI.
• Proven experience in LLM post-training, including SFT, RLHF, RLAIF, Reward Modeling, Chain-of-thought, and 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 focus on 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 models.
• Strong product intuition and ownership.

At Apple, base pay is part of our total compensation and depends on skills, qualifications, experience, and location. The range for this role is $212,000 to $386,300.

Employees can participate in stock programs, receive benefits such as medical, dental, retirement, discounts, and educational reimbursements. This role may also be eligible for bonuses, commissions, and relocation assistance. Learn more about Apple Benefits.

Apple is an equal opportunity employer committed to inclusion and diversity. We promote equal opportunity regardless of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or other protected characteristics.