Sitemap

A list of all the posts and pages found on the site. For you robots out there is an XML version available for digesting as well.

Page Not Found

Page not found. Your pixels are in another canvas. Read more

Babak Rahmani

Efficient and agentic AI (reasoning, long-horizon generalization) Read more

Page not in menu

This is a page not in th emain menu Read more

Jupyter notebook markdown generator

Posts

Debugging CWMs: Where Traces Help (and Why They Fail) Permalink

less than 1 minute read

Published: February 01, 2026

Natural image synthesis for the retina with variational information bottleneck representation

8 minute read

Published: November 19, 2022

Variational framework for partially-measured physical system control

less than 1 minute read

Published: December 01, 2021

portfolio

Portfolio item number 1

Published: February 20, 2026

Short description of portfolio item number 1
Read more

Portfolio item number 2

Published: February 20, 2026

Short description of portfolio item number 2
Read more

publications

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)

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)

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)

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

Babak Rahmani, Demetri Psaltis, Christophe Moser

NeurIPS Proceedings (2022)

Backpropagation-free Training of Deep Physical Neural Networks Internal page

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

Science (2023)

An actor-model framework for visual sensory encoding Internal page

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

Nature Communications (2024)

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)

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)

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)

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)