Jiajun Zhu 「N.B」 My Chinese name is "祝家俊".
It’s also fine to call me Jun.

I am currently a PhD student at University of Texas at Austin, advised by Atlas Wang and Aditya Akella. I previouly graduated from Zhejiang University with a major in Mathematics and a minor in Entrepreneurship. I have been working on explainable AI and large language model architectures, with my recent research focusing on diffusion language models. Beyond this, I am eager to advance trustworthiness (e.g., robustness, safety, alignment) and efficiency (e.g., MLSys, efficient architectures), and I welcome opportunities for collaboration.

Email: jiajun [at] cs [dot] utexas [dot] edu

CV  /  Google Scholar  /  Github

「News」
「Research」
TAPE
Rethinking Addressing in Language Models via Contextualized Equivariant Positional Encoding
Jiajun Zhu*, Peihao Wang*, Ruisi Cai, Jason D. Lee, Pan Li, Zhangyang Wang
ICML 2025.
Paper Code
benchmark
Towards Unveiling Sensitive and Decisive Patterns in Explainable AI with a Case Study in Geometric Deep Learning
Jiajun Zhu, Siqi Miao, Rex Ying, Pan Li
Nature Machine Intelligence.
Paper Code
「Extra」

I aspire to be a compassionate yet influential individual. In addition to my academic interest in AI, I am fascinated by other cutting-edge technologies such as brain-computer interfaces and blockchain, the inherent traits of successful entrepreneurs, and practical and effective ways to promote social good. My role models include Charlie Munger, Stephen Curry, Jian Li (李健), and many of my kind and intelligent peers.

Website template: Jon Barron's and Yunzhi Zhang's