And the TPU is a story of its own when it comes to usability. Take a look at their docs. TPU actually has to run on a separate machine, which you need to provision manually using their `ctpu` tool, and training of a trivial network on CIFAR looks like this: https://github.com/tensorflow/tpu/blob/master/models/experim...
I am a complete noob, so i just did Keras (to TF) and GPUs.
If i switch to TPU - its still the same code.
When i get the budget to cluster, i think I am better off with one of the Deepmind libs which TF based.
No?
As to deployment, once you figure out what works, there's also libtorch, or if that doesn't work for you for whatever reason, implementing a model you know working hyperparameters for can be done in a day or two on whatever framework works for your backend, including TF.