Publications

2026

  1. Under Review
    The Geometry of Ignorance: LLMs Know When to Temper Bayesian Priors
    Toni J.B. Liu, Jiajun Bao, Yizhou Liu, Gurbir Arora, Nicolas Boullé, Raphaël Sarfati, and Christopher J. Earls
    arXiv preprint, 2026
  2. Under Review
    A Graph Signal Processing Perspective on Numerical Sequence Representations in LLM In-Context Learning
    Jiajun Bao, Zihao Qi, Toni J.B. Liu, Gurbir Arora, Raphaël Sarfati, Nicolas Boullé, and Christopher J. Earls
    arXiv preprint, 2026
  3. Text-Trained LLMs Can Zero-Shot Extrapolate PDE Dynamics, Revealing a Three-Stage In-Context Learning Mechanism
    Jiajun Bao, Nicolas Boullé, Toni J.B. Liu, Raphaël Sarfati, and Christopher J. Earls
    ICLR 2026 Workshop on AI & PDEOral (5.4% oral acceptance rate), 2026
    Invited as a selected workshop contribution and published as part of the AI & PDE Focus Collection in IOP Machine Learning: Science and Technology, 2026.

2024

  1. Ensemble Kalman Filters with Resampling
    Omar Al-Ghattas, Jiajun Bao, and Daniel Sanz-Alonso
    SIAM/ASA Journal on Uncertainty Quantification, 2024