aapot
Add 780k train step and HF flax model
0ed7f5f