aapot
Add 20k train step and HF flax model
0998a82