aapot
Add 500k train step and HF flax model
8d29ba0