Publications

You can also find my articles on Google Scholar

Learning State-Tracking from Code Using Linear RNNs Internal page

Julien Siems, Riccardo Grazzi, Kirill Kalinin, Hitesh Ballani, Babak Rahmani (co-first author; supervisor)

arXiv (2026)

Debugging code world models Internal page

Babak Rahmani

arXiv (2026)

Training Deep Physical Neural Networks Internal page

Ali Momeni, Babak Rahmani, Benjamin Scellier, Logan G. Wright, Peter L. McMahon, Clara C. Wanjura, Yuhang Li, Anas Skalli, Natalia G. Berloff, Tatsuhiro Onodera, Ilker Oguz, Francesco Morichetti, Philipp del Hougne, Manuel Le Gallo, Abu Sebastian, Azalia Mirhoseini, Cheng Zhang, Danijela Marković, Daniel Brunner, Christophe Moser, Sylvain Gigan, Florian Marquardt, Aydogan Ozcan, Julie Grollier, Andrea J. Liu, Demetri Psaltis, Andrea Alù, Romain Fleury

Nature (2025)

AOC Internal page

Kirill P. Kalinin, Jannes Gladrow, Jiaqi Chu, James H. Clegg, Daniel Cletheroe, Douglas J. Kelly, Babak Rahmani (co-first author), Grace Brennan, Burcu Canakci, Fabian Falck, Michael Hansen, Jim Kleewein, Heiner Kremer, Greg O’Shea, Lucinda Pickup, Saravan Rajmohan, Ant Rowstron, Victor Ruhle, Lee Braine, Shrirang Khedekar, Natalia G. Berloff, Christos Gkantsidis, Francesca Parmigiani, Hitesh Ballani

Nature (2025)

Implicit Language Models are RNNs: Balancing Parallelization and Expressivity Internal page

Mark Schöne, Babak Rahmani (co-first author), Heiner Kremer, Fabian Falck, Hitesh Ballani, Jannes Gladrow

ICML (2025)

Regularizing the Infinite: Improved Generalization Performance with Deep Equilibrium Models Internal page

Babak Rahmani, Jannes Gladrow, Kirill Kalinin, Heiner Kremer, Christos Gkantsidis, Hitesh Ballani

NeurIPS 2024 Workshop: Machine Learning with New Compute Paradigms (MLNCP) (2024)

An actor-model framework for visual sensory encoding Internal page

Franklin Leong, Babak Rahmani, Demetri Psaltis, Christophe Moser, Diego Ghezzi

Nature Communications (2024)

Backpropagation-free Training of Deep Physical Neural Networks Internal page

Ali Momeni, Babak Rahmani, Matthieu Malléjac, Philipp del Hougne, Romain Fleury

Science (2023)

Natural image synthesis for the retina with variational information bottleneck representation Internal page

Babak Rahmani, Demetri Psaltis, Christophe Moser

NeurIPS Proceedings (2022)

Variational framework for partially-measured physical system control Internal page

Babak Rahmani, Demetri Psaltis, Christophe Moser

Fourth Workshop on Machine Learning and the Physical Sciences, NeurIPS21 (2021)

Actor neural networks for the robust control of partially measured nonlinear systems showcased for image propagation through diffuse media Internal page

Babak Rahmani, Damien Loterie, Eirini Kakkava, Navid Borhani, Uğur Teğin, Demetri Psaltis, Christophe Moser

Nature Machine Intelligence (2020)

Multimode optical fiber transmission with a deep learning network Internal page

Babak Rahmani, Damien Loterie, Georgia Konstantinou, Demetri Psaltis, Christophe Moser

Nature Light Science and Application (2018)