Ax Edoardo Calvello, Elizabeth Carlson, Nikola Kovachki, Michael N. Manta, Andrew M. Stuart 3/24/2026

Operator Learning for Smoothing and Forecasting

Theory for data-driven operator learning methods in smoothing and forecasting of dynamical systems and data assimilation.

Ax Erez Badash, Dan Boneh, Ilan Komargodski, Megha Srivastava 3/24/2026

Hawkeye: Reproducing GPU-Level Non-Determinism

Hawkeye system reproduces GPU matrix operations on CPU for verifiable ML without precision loss. Enables reproducible ML inference analysis.

Ax Jerry M. Huang, Lukas Schumacher, Niek Stevenson, Stefan T. Radev 3/24/2026

CogFormer: Learn All Your Models Once

Simulation-based inference framework enabling rapid neural network fitting across varying cognitive modeling assumptions and parameterizations.

Ax Tanay Varshney, Annie Surla, Michelle Xu, Gomathy Venkata Krishnan, Maximilian Jeblick, David Austin, Neal Vaidya, Davide Onofrio 3/24/2026

LLM Router: Prefill is All You Need

Router mechanism for LLMs using internal prefill activations and Encoder-Target Decoupling to route queries to best-performing models.

Ax Sing-Yuan Yeh, Yi-An Wu, Hau-Tieng Wu, Mao-Pei Tsui 3/24/2026

Accelerate Vector Diffusion Maps by Landmarks

Landmark-constrained algorithm to accelerate Vector Diffusion Maps framework for manifold learning on complex datasets.