aapot
Add 800k train step and HF flax model
e1fe9ac