aapot
Add 580k train step and HF flax model
5c836e2