Compute Gradients With Automatic Differentiation

PyTorch is a Python deep-learning framework for tensor computation, automatic differentiation, and dynamic neural networks. It supports CPU and GPU execution, including CUDA and ROCm acceleration.

AI Frameworks | Python · deep learning · tensor library