danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
82 stars 10 forks source link

Wrap bijections for converting transformed distributions to numpyro #139

Closed danielward27 closed 6 months ago

danielward27 commented 6 months ago

Now when a transformed distribution is converted to a numpyro distribution, it is a numpyro TransformedDistribution, made by seperately wrapping the base distribution and the bijection to the numpyro types. This e.g. facilitates reparameterization of numpyro model.