search

poets-ai / elegy

A High Level API for Deep Learning in JAX
https://poets-ai.github.io/elegy/
MIT License
470 stars 32 forks source link

Run Example in TPUs #29

Open charlielito opened 4 years ago

charlielito commented 4 years ago

Create an example using TPUs for training. IT can be in Colab

cgarciae commented 4 years ago

@charlielito this is almost done right?

charlielito commented 4 years ago

@charlielito this is almost done right?

Yeah. With the MNIST Example. But we were discussing maybe including a more complex example for TPU?