aapot
Add 520k train step and HF flax model
41d55d8