Google har publisert TorchTPU som åpen kildekode. Det er en egen PyTorch-backend for selskapets TPU-brikker.

Kilde: https://github.com/google-pytorch/torch_tpu