Meta Open Source
Project Details


Tensors and Dynamic neural networks in Python with strong GPU acceleration

Hero image for PyTorch

Project Overview

PyTorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment.

PyTorch has the following features:

  1. Production Ready: Transition seamlessly between eager and graph modes with TorchScript, and accelerate the path to production with TorchServe.
  2. Distributed Training: Scalable distributed training and performance optimization in research and production is enabled by the torch.distributed backend.
  3. Robust Ecosystem: A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and on the server using Node and power mobile apps using React Native.
  4. Cloud Support: PyTorch is well supported on major cloud platforms, providing frictionless development and easy scaling.