Codebase as it was on the date of the final revisions for the paper Action abstractions for amortized sampling by
Oussama Boussif, Léna Néhale Ezzine, Joseph D Viviano, Michał Koziarski, Moksh Jain, Nikolay Malkin, Emmanuel Bengio, Rim Assouel, Yoshua Bengio published at ICLR 2025.