Closed torfjelde closed 2 months ago
@yebai @devmotion should be a quick merge
The PR made me wonder whether it would be possible to improve performance of with_logabsdet_jacobian for SimplexBijector by not performing transform and logabsdetjac separately when both are requested. Doesn't block this PR and maybe would lead to more code duplications though.
Definitively! But yeah, probs for another PR (I'll make issue) :+1:
Related: https://github.com/TuringLang/Turing.jl/issues/2190