issues
search
google
/
flax
Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.79k
stars
609
forks
source link
Cannot load checkpoint saved in flax 0.5.3 with flax 0.6.1
#3977
Closed
tpgh24
closed
3 weeks ago