ehoogeboom/multinomial_diffusion