Jiaxuan Zou

Undergraduate Student @ XJTU | Research Intern @ GSAI, RUC

Jiaxuan Zou

School of Mathematics and Statistics Xi'an Jiaotong University

The best way to reach me is via email.

Hi, I am Jiaxuan Zou (邹嘉轩).

I am an undergraduate student in Mathematics and Statistics at Xi’an Jiaotong University and a research intern at the Gaoling School of Artificial Intelligence, Renmin University of China, advised by Prof. Yong Liu.

My research focuses on mechanistic interpretability, deep learning theory, optimization, and scaling laws. I am especially interested in turning empirical training phenomena into first-principles mechanisms: why models train, where behaviors emerge, and when scaling or optimization rules break.

I also work as an AI Technical Consultant for a Tsinghua-affiliated AI startup developing “AI + K-12 Education” products with the Beijing Dongcheng District Education Commission.

I write research notes on my blog and welcome conversations on these topics. I am currently seeking industry research internships in LLM pre-training and AI theory, as well as PhD positions for Fall 2028. More background: English / 中文.

Research Interests

  • Mechanistic interpretability of LLMs
  • Training dynamics of finite-width networks
  • Optimizer design for LLM pre-training
  • Scaling laws and their failure modes
  • Deep learning theory and optimization

News

May 25, 2026 Our Nora optimizer has been included in the ScalingOpt optimizer library.
Apr 18, 2026 I joined the ScalingOpt project as a co-maintainer, working on optimizer design for large language model training.

Latest Posts

Selected Publications

  1. Kaczmarz Linear Attention
    Kaczmarz Linear Attention
    Jiaxuan Zou, Ruifeng Ren, and Yong Liu
    2026
  2. Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer
    Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer
    Jinghui Yuan, Jiaxuan Zou, Shuo Wang, and 2 more authors
    2026
  3. Effective Frontiers: A Unification of Neural Scaling Laws
    Effective Frontiers: A Unification of Neural Scaling Laws
    Jiaxuan Zou, Zixuan Gong, Ye Su, and 2 more authors
    Under Review, 2026
  4. Capabilities and Fundamental Limits of Latent Chain-of-Thought
    Capabilities and Fundamental Limits of Latent Chain-of-Thought
    Jiaxuan Zou, Yaozhong Xiong, and Yong Liu
    Under Review, 2026
  5. Statistical MIA: Rethinking Membership Inference Attack for Reliable Unlearning Auditing
    Statistical MIA: Rethinking Membership Inference Attack for Reliable Unlearning Auditing
    Jialong Sun, Zeming Wei, Jiaxuan Zou, and 5 more authors
    Under Review, 2026
  6. FreIE: Low-Frequency Spectral Bias in Neural Networks for Time-Series Tasks
    FreIE: Low-Frequency Spectral Bias in Neural Networks for Time-Series Tasks
    Jialong Sun, Xinpeng Ling, Jiaxuan Zou, and 2 more authors
    In IEEE ICDM, 2025