aapot
Add 700k train step and HF flax model
1643cbd