Add Flax checkpoints

#95
by ksmcg - opened
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment