axlearn

checkpointing policy at last iteration (#52)

Adds read_state_spec to checkpointer.py. (#49)

Lists supported bundler types when reporting the unknown bundler type error (#50)

Adds Module.vprint, which uses jax.debug.print to print tensor contents (#48)

fix alibi logits bias order (#37)