aapot
Add 720k train step and HF flax model
73866d2