Simple MNIST training example

yixiaoer / tpux

A set of Python scripts that makes your experience on TPU better

MIT License

40 stars 0 forks source link

Open s-smits opened 3 months ago

s-smits commented 3 months ago

I think it would be helpful to add a MNIST example with JAX through podrun.

yixiaoer commented 3 months ago

Hey, thanks for the feedback! The MNIST training example will be added later. For now, you can refer to the tpu-training-example using podrun.