/Discrete-Time-Diffusion-Models-for-Discrete-Data

Unofficial PyTorch reimplementations of the papers Structured Denoising Diffusion Models in Discrete State-Spaces by J. Austin et al. and Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions by E. Hoogeboom et al.

Primary LanguagePythonMIT LicenseMIT

Stargazers