aapot
Add 880k train step and HF flax model
f4313c8