Austin Kaiji Wang

Research Scientist @ ByteDance Seed. Caltech CS.

wukang-pfp.jpg

Palo Alto, CA

Hi! I’m Austin, a Caltech CS graduate and researcher at ByteDance Seed, where I currently work on post-training for visual foundation models. During my undergrad, I was fortunate to do research in the labs of Prof. Yisong Yue from Caltech and Prof. Stefano Ermon from Stanford.

My research focuses on developing efficient and principled algorithms that unlock new capabilities from generative models. I am particularly interested in post-training, inference-time scaling, and controllable generation: how to adapt powerful models to better satisfy rewards, constraints, and downstream objectives. More broadly, I aim to design methods that make generative models more capable and accessible across creative and scientific applications.

news

Jul 20, 2026 Joined ByteDance Seed.
Jun 12, 2026 Graduated from Caltech!
Jan 25, 2026 SDPO was accepted to ICLR 2026!

selected publications

  1. Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models
    Austin Wang, Jiaqi Han, Stefano Ermon, and 1 more author
    arXiv preprint arXiv:2605.26491, 2026
  2. Discrete diffusion trajectory alignment via stepwise decomposition
    Jiaqi Han*, Austin Wang*, Minkai Xu, and 6 more authors
    International Conference on Learning Representations (ICLR 2026), 2026
  3. Blade: A derivative-free bayesian inversion method using diffusion priors
    Hongkai Zheng*, Austin Wang*, Zihui Wu, and 3 more authors
    arXiv preprint arXiv:2510.10968, 2025