Metadata-Version: 2.4
Name: cutensor-cu12
Version: 2.3.0
Summary: NVIDIA cuTENSOR
Home-page: https://developer.nvidia.com/cutensor
Author: NVIDIA Corporation
Author-email: cuda_installer@nvidia.com
License: NVIDIA Proprietary Software
Keywords: cuda,nvidia,machine learning,tensor network,high-performance computing
Classifier: Topic :: Scientific/Engineering
Classifier: Environment :: GPU :: NVIDIA CUDA
Classifier: Environment :: GPU :: NVIDIA CUDA :: 12
Description-Content-Type: text/x-rst
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: keywords
Dynamic: license
Dynamic: summary

###############################################################
cuTENSOR: A High-Performance CUDA Library For Tensor Primitives
###############################################################

`cuTENSOR <https://developer.nvidia.com/cutensor>`_ is a high-performance CUDA library for tensor primitives.

Key Features
============

* Extensive mixed-precision support:

  * FP64 inputs with FP32 compute.
  * FP32 inputs with FP16, BF16, or TF32 compute.
  * Complex-times-real operations.
  * Conjugate (without transpose) 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.

    * Support just-in-time compilation of dedicated kernels.

  * Tensor reductions (including partial reductions).
  * Element-wise tensor operations:

    * Support for various activation functions.
    * Support for padding of the output tensor
    * Arbitrary tensor permutations.
    * Conversion between different data types.

Documentation
=============

Please refer to https://docs.nvidia.com/cuda/cutensor/index.html for the cuTENSOR documentation.

Installation
============

The cuTENSOR wheel can be installed as follows:

.. code-block:: bash

   pip install cutensor-cuXX

where XX is the CUDA major version (currently CUDA 12 & 13 are supported).
The package ``cutensor`` (without the ``-cuXX`` suffix) is deprecated. If you have
``cutensor`` installed, please remove it prior to installing ``cutensor-cuXX``.
