aapot
Add 820k train step and HF flax model
8fe61f1