Alternative kind of continuous relaxation of a discrete diffusion

[murrellb]

That might be easy to implement in this framework: https://arxiv.org/pdf/2309.02530.pdf