Open SamDuffield opened 3 weeks ago
We added associative scan to sample in #30
sample
But we still need to do so for thermox.prob.log_prob_identity_diffusion
thermox.prob.log_prob_identity_diffusion
This shouldn't be too hard, but will need to think about how to keep memory usage low in the associative framework
We added associative scan to
sample
in #30But we still need to do so for
thermox.prob.log_prob_identity_diffusion
This shouldn't be too hard, but will need to think about how to keep memory usage low in the associative framework