aapot
Add 750k train step and HF flax model
d1bc6bc