danielward27 / flowjax

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

Filter spec #32

Closed danielward27 closed 2 years ago

danielward27 commented 2 years ago

Add option to provide a filter_spec to specify trainable parameters.