PyTorch documentation — PyTorch 1.10.1 documentation
https://pytorch.org/docsPyTorch documentation. PyTorch is an optimized tensor library for deep learning using GPUs and CPUs. Features described in this documentation are classified by release status: Stable: These features will be maintained long-term and there should generally be no major performance limitations or gaps in documentation.
Google Colab
colab.research.google.com › github › pytorchRun basic PyTorch functions on TPUs, like creating and adding tensors. Run PyTorch modules and autograd on TPUs. Run PyTorch networks on TPUs. PyTorch/XLA is a package that lets PyTorch connect to Cloud TPUs and use TPU cores as devices.