search

jax-ml / bayeux

State of the art inference for your bayesian models.
https://jax-ml.github.io/bayeux/
Apache License 2.0
177 stars 8 forks source link

Add pymc example. #19

Closed copybara-service[bot] closed 10 months ago

copybara-service[bot] commented 10 months ago

Add pymc example.

This turned up a mistake in the TFP vi implementation, where the log density needs to be vmapped.