PyTorch implementation of the paper "Entropy-MCMC: Sampling from Flat Basins with Ease" (ICLR 2024)
Primary LanguagePython