jax_md/_nn/uma/checkpoints/*.npz filter=lfs diff=lfs merge=lfs -text
