tensor-library

PyTorch is an open-source deep learning framework providing GPU-accelerated tensor computation and dynamic neural networks via tape-based autograd. The GitHub README offers installation instructions, component overviews, and links to documentation.