aapot
Add 480k train step and HF flax model
7a31b62