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.12.0
Required dependencies:
cuda-bindings
|
cuda-toolkit
|
filelock
|
fsspec
|
jinja2
|
networkx
|
nvidia-cublas
|
nvidia-cudnn-cu13
|
nvidia-cusparselt-cu13
|
nvidia-nccl-cu13
|
nvidia-nvshmem-cu13
|
setuptools
|
sympy
|
triton
|
typing-extensions
Optional dependencies:
opt-einsum
|
optree
|
pyyaml
Downloads last day:
3,086,118
Downloads last week:
19,840,156
Downloads last month:
85,991,620