Support for TPU (in Colab or otherwise) #37
Description
Google offers access to cloud TPUs via Colab, but using TPUs requires some modification to existing Tensorflow/Keras code to get any actual speed-ups. Details on that process can be found at this link: https://www.tensorflow.org/guide/using_tpu.
This is likely a low-priority item, as training on medical data using Colab is already fraught because of privacy concerns, but may also be not too difficult to implement in the context of DeepNeuro. Likely an optional "tpu" parameter would be added to keras_model.py and tensorflow_model.py, along with a few extra lines of code to convert models to their respective TPU modes.