aapot
Add 650k train step and HF flax model
77e0b25