aapot
Add 920k train step and HF flax model
d5f3c4c