Flax is a neural network library for JAX that is designed for flexibility.
5.8k
stars
613
forks
source link
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file. #4055
Closed
copybara-service[bot] closed 1 day ago
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.