PyPI page
Home page
Author: None
License: BSD-3-Clause
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Latest version: 2.9.1
Required dependencies: filelock | fsspec | jinja2 | networkx | nvidia-cublas-cu12 | nvidia-cuda-cupti-cu12 | nvidia-cuda-nvrtc-cu12 | nvidia-cuda-runtime-cu12 | nvidia-cudnn-cu12 | nvidia-cufft-cu12 | nvidia-cufile-cu12 | nvidia-curand-cu12 | nvidia-cusolver-cu12 | nvidia-cusparse-cu12 | nvidia-cusparselt-cu12 | nvidia-nccl-cu12 | nvidia-nvjitlink-cu12 | nvidia-nvshmem-cu12 | nvidia-nvtx-cu12 | setuptools | sympy | triton | typing-extensions
Optional dependencies: opt-einsum | optree | pyyaml
Downloads last day: 1,964,969
Downloads last week: 16,225,407
Downloads last month: 71,903,054