Add TFP NUTS and HMC samplers. - Githubissues

jax-ml / bayeux

State of the art inference for your bayesian models.

https://jax-ml.github.io/bayeux/

Apache License 2.0

162 stars 6 forks source link

Add TFP NUTS and HMC samplers. #26

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

Add TFP NUTS and HMC samplers.

Also clean up keyword argument handling.