aapot
Add 600k train step and HF flax model
a4de590