Paper accepted at ICLR 2025