aapot
Add 280k train step and HF flax model
2f785fb