python-pytorch 1.9.0 Python library for tensor computation and deep neural networks

PyTorch is a Python package that provides two high-level features:

  • tensor computation (like NumPy) with strong GPU acceleration;

  • deep neural networks (DNNs) built on a tape-based autograd system.

You can reuse Python packages such as NumPy, SciPy, and Cython to extend PyTorch when needed.

Note: currently this package does not provide GPU support.