PyTorch is an open source, machine learning framework based on Python. It enables you to perform scientific and tensor computations with the aid of graphical processing units (GPUs). You can use it to develop and train deep learning neural networks using automatic differentiation (a calculation process that gives exact values in constant time).
Key features of PyTorch include:
Simple interface—includes an easy to use API that can be used with Python, C++, or Java.
Pythonic in nature—integrates smoothly with the Python data science stack and enables you to leverage Python services and functionalities.
Computational graphs—includes capabilities for dynamic computational graphs that you can customize during runtime.