search

danielward27 / flowjax

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

Pass params in step #111

Closed danielward27 closed 1 year ago

danielward27 commented 1 year ago

This is needed for some optimizers e.g. if they include L2 regularisation.