AIML Researcher/Engineer - Foundation Model Post-Training

Apple Inc

Actively hiring Posted this week
Cupertino, CA Posted 4 days ago
View original post Log in to save

At a glance

AI generated

TL;DR

As an AIML Researcher/Engineer at Apple’s Foundation Model team, you will contribute to the development of large-scale foundation models by designing and iterating on post-training strategies that enhance model capabilities in areas such as instruction following, tool use, and deep reasoning. Your daily tasks include pioneering novel algorithms for preference optimization and safety, driving data strategy through high-quality synthetic data generation, and developing robust evaluation methodologies to measure real-world performance accurately. You will collaborate closely with pre-training teams on architecture choices and work with product teams to translate user requirements into model functionalities. This role requires expertise in deep learning, particularly LLMs and post-training techniques, proficiency in Python and frameworks like JAX or PyTorch, and experience with distributed training challenges at scale.

Skills

Python PyTorch JAX Reinforcement Learning LLMs Distributed Training Transformers Curriculum Learning Evaluation Methodologies Deep Learning Data Generation Automated Data Filtering

What you'll do

  • Design and iterate on end-to-end post-training strategies to unlock model capacities.
  • Pioneer novel algorithms for preference optimization, model steering, and safety.
  • Drive data strategy by researching methods for high-quality human and synthetic data generation.
  • Develop robust evaluation methodologies to measure model helpfulness and factuality.
  • Collaborate with pre-training teams to inform architecture choices and align user requirements.

What we're looking for

  • Demonstrated expertise in deep learning, LLMs, post-training, or reinforcement learning.
  • Proficient programming skills in Python and major deep learning frameworks like JAX or PyTorch.
  • Masters/PhD in Computer Science, Machine Learning, or a related technical field.
  • Experience training large models at scale with knowledge of distributed training challenges.
  • Ability to design robust evaluation methodologies for model performance measurement.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 255 open roles on FindRole.

Listed pay typically runs $171,600–$272,100 across 182 roles with salary data.

Most-posted roles

View all roles at Apple Inc