aapot
Add 850K train step and HF flax model
06dc3e1