A State Representation for Diminishing Rewards. - Ted Moskovitz; Samo Hromadka; Ahmed Touati; Diana Borsa; Maneesh Sahani (NeurIPS, 2023)

Fast unsupervised ground metric learning with tree-Wasserstein distance. - Kira Michaela Düsterwald; Samo Hromadka; Makoto Yamada (ICLR, 2025)

Maximum Likelihood Learning of Latent Dynamics Without Reconstruction. - Samo Hromadka; Kai Biegun; Lior Fox; James Heald; Maneesh Sahani (CoRR, 2025)