issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.63k
stars
174
forks
source link
allow wrappers.flatten to work with 0-dim arrays
#816
Closed
copybara-service[bot]
closed
6 months ago
copybara-service[bot]
commented
6 months ago
allow wrappers.flatten to work with 0-dim arrays
allow wrappers.flatten to work with 0-dim arrays