aapot
Add 380k train step and HF flax model
916b5f4