
Learning from experience, at frontier scale.
Core contributor to RL fundamentals at xAI (now SpaceX Super Intelligence): the science of RL training behind the Grok RL big runs, including one of the longest stable RL runs to date. Previously led RL stability and efficiency research at ByteDance.
PhD, The Chinese University of Hong Kong (2025), advised by Prof. Zhi-Quan (Tom) Luo, with Prof. Benjamin Van Roy (Stanford & DeepMind) on my thesis committee.
First published in September 2025. Our fix, masked importance sampling (MIS), made principled as Trust Region Masking (TRM), is adopted, shipped or discussed by:
Core algorithms and training recipes, optimizer behavior and numerics at scale, and first-principles theory of how an update moves a model and how training should scale.
Rollout Correction is veRL’s framework for off-policy training, and MIS and TRM ship in it.
Dr. Kernel-14B writes Triton kernels and refines them using execution feedback. Its multi-turn RL pairs an unbiased turn-level advantage with our mismatch correction against early collapse. With test-time scaling, 31.6% of its KernelBench Level 2 kernels run at least 1.2× faster than PyTorch, ahead of Claude-4.5-Sonnet (26.7%) and GPT-5 (28.6%).
Human-level Atari with 15% of DDQN’s interaction data, and Thompson-sampling regret with only Θ(d log T) ensemble directions.
I started RL-Seminar, an RL theory seminar, in 2019 and ran it through 2023.
Open to conversations on research collaboration, advising, and speaking. I read every email and write back.