danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
101 stars 14 forks source link

Planar flow #96

Closed danielward27 closed 1 year ago

danielward27 commented 1 year ago

Add planar flow. In the unconditional case, the parameters are learned directly, in the conditional case an MLP conditioner is used to output the parameters.