aapot
Add 1200k train step and HF flax model
ef4601c