aapot
Add 1050k train step and HF flax model
3bc5dc7