Ax David P. Morton, Oscar Dowson, Bernardo K. Pagnoncelli 4/9/2026

MDP modeling for multi-stage stochastic programs

MDP modeling framework extending policy graphs for multi-stage stochastic programs with decision-dependent uncertainty and statistical learning.

Ax Max Hopkins, Russell Impagliazzo, Christopher Ye 4/9/2026

Approximate Replicability in Learning

Approximate replicability framework for machine learning algorithms that remain stable under input resampling.

Ax James O'Neill, Robert Clancy, Mariia Matskevichus, Fergal Reid 4/9/2026

Low-Rank Key Value Attention

Low-Rank Key-Value (LRKV) attention reduces transformer KV cache memory by exploiting redundancy across attention heads with low-rank residuals.

Ax Alex Morehead, Miruna Cretu, Antonia Panescu, Rishabh Anand, Maurice Weiler, Tynan Perez, Samuel Blau, Steven Farrell, Wahid Bhimji, Anubhav Jain, Hrushikesh Sahasrabuddhe, Pietro Lio, Tommi Jaakkola, Rafael Gomez-Bombarelli, Rex Ying, N. Benjamin Erichson, Michael W. Mahoney 4/9/2026

Zatom-1: A Multimodal Flow Foundation Model for 3D Molecules and Materials

Open-source foundation model for 3D molecular and materials modeling with both generative and predictive capabilities.

Ax Chenxu Yang, Chuanyu Qin, Qingyi Si, Minghui Chen, Naibin Gu, Dingyu Yao, Zheng Lin, Weiping Wang, Jiaqi Wang, Nan Duan 4/9/2026

Self-Distilled RLVR

On-policy self-distillation approach for LLM training combining dense teacher signals with sparse verifiable rewards from environment feedback.

Ax Tijana Zrnic, Emmanuel J. Cand\`es 4/9/2026

Active Statistical Inference

Active inference methodology for ML-assisted data collection, using models to identify which points merit labeling under budget constraints for efficient learning.

Ax Timo Gierlich, Andreas Baumbach, Akos F. Kungl, Kevin Max, Mihai A. Petrovici 4/9/2026

Spike-based alignment learning solves the weight transport problem

Spike-based alignment learning resolves weight transport problem in neural networks, enabling local computation compatible with biological networks and neuromorphic hardware.

Ax Andrea Montanari, Viet Vu 4/9/2026

Computational bottlenecks for denoising diffusions

Analyzes computational bottlenecks in denoising diffusion models, examining efficiency of drift learning and sampling procedures for probability distribution approximation.