github.com
PyTorch is a Python package providing tensor computation with strong GPU acceleration and deep neural networks built on a tape-based autograd system. It includes components such as torch, torch.autograd, torch.nn, and torch.utils, and can be used as a NumPy replacement or a deep learning research platform.