############################################################### cuTENSOR: A High-Performance CUDA Library For Tensor Primitives ###############################################################
cuTENSOR <https://developer.nvidia.com/cutensor>_ is a high-performance CUDA library for tensor primitives.
Extensive mixed-precision support:
Support for up to 64-dimensional tensors.
Arbitrary data layouts.
Trivially serializable data structures.
Main computational routines:
Direct (i.e., transpose-free) tensor contractions.
Tensor reductions (including partial reductions).
Element-wise tensor operations:
Please refer to https://docs.nvidia.com/cuda/cutensor/index.html for the cuTENSOR documentation.
The cuTENSOR wheel can be installed as follows:
.. code-block:: bash
pip install cutensor-cuXX
where XX is the CUDA major version (currently CUDA 11 & 12 are supported).
The package cutensor (without the -cuXX suffix) is deprecated. If you have
cutensor installed, please remove it prior to installing cutensor-cuXX.
FAQs
NVIDIA cuTENSOR
We found that cutensor-cu12 demonstrated a healthy version release cadence and project activity because the last version was released less than a year ago. It has 4 open source maintainers collaborating on the project.
Did you know?
Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.
Security News
PyPI confirms no security flaws were exploited in the Ultralytics supply chain attack and highlights improvements for safer package publishing.
Security News
Research
The Socket Research Team breaks down a malicious wrapper package that uses obfuscation to harvest credentials and exfiltrate sensitive data.
Research
Security News
Attackers used a malicious npm package typosquatting a popular ESLint plugin to steal sensitive data, execute commands, and exploit developer systems.
