danielward27 / flowjax

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

Tidy #49

Closed danielward27 closed 1 year ago

danielward27 commented 1 year ago

Refactoring will have caused some breaking changes. Masks and neural networks now have their own modules (masks.py and nn respectively).