aapot
Add 900k train step and HF flax model
12e012f