aapot
Add 1000K train step and HF flax model
ba320be