danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
101 stars 14 forks source link

Switch sample args #68

Closed danielward27 closed 1 year ago

danielward27 commented 1 year ago

Switch condition and sample_shape arguments in Distribution.sample and Distribution.sample_and_log_prob.