Numpyro conditional bug fix

danielward27 / flowjax

https://danielward27.github.io/flowjax/

MIT License

82 stars 10 forks source link

Numpyro conditional bug fix #136

Closed danielward27 closed 7 months ago

danielward27 commented 7 months ago

Fixes bug where when wrapping a flowjax distribution into a numpyro distribution, the base distribution sample shape did not expand appropriately when a batch of conditioning variables was used.