aapot's picture
Add 500k train step and HF flax model
aee73fc