danielward27 / flowjax

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

Add bisection inverter #122

Closed danielward27 closed 9 months ago

danielward27 commented 9 months ago

Primary benefit of this is that it allows us to 1) Sample block neural autoregressive flows fitted with maximum likelihood, and 2) evaluate the log probability of block neural autoregressive flows fitted with variational inference. This can potentially also be used in other flows in the future.