Flax is a neural network library for JAX that is designed for flexibility.
5.78k
stars
610
forks
source link
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file. #4039
Open
copybara-service[bot] opened 5 days ago
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.