Create README.md

Saving weights and logs of step 2250

Saving weights and logs of step 2250

Updated events

Saving weights and logs of step 750

Saving weights and logs of step 750

Restarted training

Restarted training

Saving weights and logs of step 5250

Saving weights and logs of step 5250

Saving weights and logs of step 4500

Saving weights and logs of step 4500

Saving weights and logs of step 3750

Saving weights and logs of step 3000

Saving weights and logs of step 2250

Saving weights and logs of step 2250

Saving weights and logs of step 1500

Saving weights and logs of step 750

Saving weights and logs of step 750

250 steps

After a few hours of training

Epoch 01

Testing Jax model training. After a few iterations (still in the first epoch), testing model save

Testing JAX training with TPU VM, After 1 Epoch

initial commit