aapot
Add 520k train step and HF flax model
e26543b