Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)
Core
Train large language models (LLMs) using reinforcement learning and develop methods to improve out-of-distribution generalization.
Role type
Senior IC machine-learning engineer (reinforcement learning & generalization)
Builds
Production-ready training recipes and robust LLMs for enterprise customers
Domain
Generative AI / Large Language Models
Deliverable
production ML models
Required skills
Reinforcement learning (PPO, GRPO, RLHF, RLVR), LLM training, experimental design, distributed training, reward design, generalization research
Preferred skills
JAX, PyTorch, FSDP, Megatron
Technologies
JAX, PyTorch, FSDP, Megatron
Responsibilities
Design and analyze large-scale RL post-training experiments; Research methods for out-of-distribution generalization; Build held-out evaluations to catch reward hacking; Diagnose training dynamics and failure modes; Convert successful research into reproducible production recipes
Seniority
Senior, hands-on IC
