Developed by Meta, PyTorch is a popular machine learning library that helps develop and train neural networks.

PyTorch 1.10 is production ready, with a rich ecosystem of tools and libraries for deep learning, computer vision, natural language processing, and more. Here's how to get started with PyTorch.

Although PyTorch has been lagging behind TensorFlow and JAX in XLA/TPU support, the situation has improved greatly as of 2022.

Other new improvements in PyTorch 1.8.1 include enabling autocast for PyTorch xla, making the torch. submodule import more autocomplete-friendly, and more.

Courtesy of PyTorch/XLA (a now generally available Python library), Google's Cloud TPUs can better support Facebook's PyTorch machine learning framework.

That framework helps PyTorch users “find bottlenecks and adapt their programs to run more efficiently on cloud TPUs,” said Craig Wiley, director of product development for Google’s Cloud AI platform.

PyTorch / XLA combines the intuitive APIs of PyTorch with the strengths of the XLA linear algebra compiler, which can target CPUs, GPUs, and Cloud TPUs, including Cloud TPU Pods. PyTorch / XLA will ...