Add associative scan to `log_prob`

[SamDuffield]

We added associative scan to sample in #30

But 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