Austin Kaiji Wang
Research Scientist @ ByteDance Seed. Caltech CS.
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
- Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion ModelsarXiv preprint arXiv:2605.26491, 2026
- Discrete diffusion trajectory alignment via stepwise decompositionInternational Conference on Learning Representations (ICLR 2026), 2026
- Blade: A derivative-free bayesian inversion method using diffusion priorsarXiv preprint arXiv:2510.10968, 2025