danielward27 / flowjax

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

Simplify wrappers #175

Closed danielward27 closed 1 month ago

danielward27 commented 1 month ago

Adds some examples, and slightly simplifies the code. Small breaking change, as the method recursive_unwrap is removed. For unwrapping all unwrappables flowjax.wrappers.unwrap should be used.