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.

Source: PyTorch.org