aapot
Add 720k train step and HF flax model
f39193d