PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing. The Python-based scientific computing package serves two broad purposes:

  • A replacement for NumPy to use the power of GPUs and other accelerators.
  • An automatic differentiation library that is useful to implement neural networks.