search

danielward27 / flowjax

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

Use math instead of jnp in TanhLinearTails init #59

Closed danielward27 closed 1 year ago