LogoAIAny
Icon for item

PyTorch

Open-source deep-learning framework for building, training and deploying neural networks on GPUs and CPUs.

Introduction

What is PyTorch?

PyTorch is an open-source deep-learning library originally developed by Meta AI and now governed by the PyTorch Foundation under the Linux Foundation umbrella. It offers a Pythonic, imperative interface with dynamic computation graphs that make research prototyping fast and intuitive.

Key capabilities
  • Tensors & Autograd – NumPy-like tensors with GPU acceleration and a tape-based automatic-differentiation engine.
  • TorchScript & TorchDynamo – Tools to turn eager Python code into optimized graphs for production workflows, introduced with PyTorch 2.0 for up-to-2× speed-ups.
  • Distributed & Accelerated Training – Native primitives for data-parallel and model-parallel training across multiple GPUs or nodes, plus seamless CUDA/ROCm support.
  • Rich Ecosystem – Hundreds of domain libraries (TorchVision, TorchAudio, TorchText, PyTorch Lightning, etc.) and community projects for vision, NLP, graphs, RL and more.
  • Production-ready Serving – TorchServe, ONNX export, mobile runtimes and integrations with every major cloud platform.
Adoption & Governance

PyTorch powers research and production systems at organisations such as Tesla, Uber and Salesforce, and underpins open-source projects like Hugging Face Transformers. It is released under the BSD-3-Clause license and stewarded by the PyTorch Foundation, ensuring vendor-neutral governance and a vibrant contributor community.

Information

  • Websitepytorch.org
  • AuthorsMeta AI, Adam Paszke, Sam Gross, Soumith Chintala, Gregory Chanan
  • Published date2016/09/01

Categories