aapot
Add 980k train step and HF flax model
2c88aa3