search

danielward27 / flowjax

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

Triangular weight norm #44

Closed danielward27 closed 2 years ago

danielward27 commented 2 years ago

Add weight normalisation for TraingularAffine. Also export StandardNormal (a standard normal distribution with no trainable parameters).