Yeah I'd assume this is to allow us to train on their TPUs with torch instead of needing to rent a GPU