Authors

Aayush Mishra

Publications 2

Preprint Open access

Enabling Dynamic Computation in Looped LMs

Aayush Mishra, Arnau Padrés Masdemont, Victor Conchello Vendrell et al. · 2026

Looped LMs are parameter efficient and promise dynamic computation (saving memory and FLOPs on easy tokens). However, state-of-the-art open Looped LMs trained with this dynamic computation capability (Ouro models) do not realize it in practice as each loop iteration (depth) requires its own level of KV-cache, necessita …

Preprint Open access

Mitigating Representation Gaps in Amortized Bayesian Inference with Auxiliary Supervision

Hans Olischläger, Svenja Jedhoff, Šimon Kucharský et al. · 2026

Casting Bayesian inference as a neural network optimization problem targeting an amortized posterior is attractive, as it extends to otherwise intractable statistical models and offers near instantaneous inference for new datasets after prepaying the training cost. Although theory guarantees faithfulness under ideal co …

Co-authors