aapot
Add 350k train step and HF flax model
88b2b43