Update loss function for MaskedDiscreteDiffusion#62

Open

Jackryd wants to merge 1 commit into

MurrellGroup:masked-discretefrom

Commits on Jul 25, 2024

Update loss function for MaskedDiscreteDiffusion to make it work on GPU, allow for batched timesteps and non-batched data
Jackryd
authored