###############################################################
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 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``.
Raw data
{
"_id": null,
"home_page": "https://developer.nvidia.com/cutensor",
"name": "cutensor-cu11",
"maintainer": "",
"docs_url": null,
"requires_python": "",
"maintainer_email": "",
"keywords": "cuda,nvidia,machine learning,tensor network,high-performance computing",
"author": "NVIDIA Corporation",
"author_email": "cuda_installer@nvidia.com",
"download_url": "",
"platform": null,
"description": "###############################################################\ncuTENSOR: A High-Performance CUDA Library For Tensor Primitives\n###############################################################\n\n`cuTENSOR <https://developer.nvidia.com/cutensor>`_ is a high-performance CUDA library for tensor primitives.\n\nKey Features\n============\n\n* Extensive mixed-precision support:\n\n * FP64 inputs with FP32 compute.\n * FP32 inputs with FP16, BF16, or TF32 compute.\n * Complex-times-real operations.\n * Conjugate (without transpose) support.\n\n* Support for up to 64-dimensional tensors.\n* Arbitrary data layouts.\n* Trivially serializable data structures.\n* Main computational routines:\n\n * Direct (i.e., transpose-free) tensor contractions.\n\n * Support just-in-time compilation of dedicated kernels.\n\n * Tensor reductions (including partial reductions).\n * Element-wise tensor operations:\n\n * Support for various activation functions.\n * Support for padding of the output tensor\n * Arbitrary tensor permutations.\n * Conversion between different data types.\n\nDocumentation\n=============\n\nPlease refer to https://docs.nvidia.com/cuda/cutensor/index.html for the cuTENSOR documentation.\n\nInstallation\n============\n\nThe cuTENSOR wheel can be installed as follows:\n\n.. code-block:: bash\n\n pip install cutensor-cuXX\n\nwhere XX is the CUDA major version (currently CUDA 11 & 12 are supported).\nThe package ``cutensor`` (without the ``-cuXX`` suffix) is deprecated. If you have\n``cutensor`` installed, please remove it prior to installing ``cutensor-cuXX``.\n",
"bugtrack_url": null,
"license": "NVIDIA Proprietary Software",
"summary": "NVIDIA cuTENSOR",
"version": "2.0.1",
"project_urls": {
"Homepage": "https://developer.nvidia.com/cutensor"
},
"split_keywords": [
"cuda",
"nvidia",
"machine learning",
"tensor network",
"high-performance computing"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "4308629a92a86b5a9c2d9587ca0de5f82f9bf5eceb3780df427ef86efd253858",
"md5": "bc524bd69850cf15f8ca10dd38d76b96",
"sha256": "c39ca2a10b541c7c1dfedd03103d145f4fdacaebfa1ed77e62307c2757390e0f"
},
"downloads": -1,
"filename": "cutensor_cu11-2.0.1-py3-none-manylinux2014_aarch64.whl",
"has_sig": false,
"md5_digest": "bc524bd69850cf15f8ca10dd38d76b96",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 143589605,
"upload_time": "2024-02-08T20:06:20",
"upload_time_iso_8601": "2024-02-08T20:06:20.531326Z",
"url": "https://files.pythonhosted.org/packages/43/08/629a92a86b5a9c2d9587ca0de5f82f9bf5eceb3780df427ef86efd253858/cutensor_cu11-2.0.1-py3-none-manylinux2014_aarch64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "ac360374fd785658de9e96258575554eeabf5606607ca18bf9439a9250db981f",
"md5": "e92503e1a7a70026218d59ba2c925a52",
"sha256": "a1777d1dba54fbab69df515717cb658f61338d550aec7cf868eb99bd091bee57"
},
"downloads": -1,
"filename": "cutensor_cu11-2.0.1-py3-none-manylinux2014_x86_64.whl",
"has_sig": false,
"md5_digest": "e92503e1a7a70026218d59ba2c925a52",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 143810828,
"upload_time": "2024-02-08T20:07:30",
"upload_time_iso_8601": "2024-02-08T20:07:30.723423Z",
"url": "https://files.pythonhosted.org/packages/ac/36/0374fd785658de9e96258575554eeabf5606607ca18bf9439a9250db981f/cutensor_cu11-2.0.1-py3-none-manylinux2014_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "4249ebeeb12e73eab21aff1249922a32e4153aae5bcccfab718ef463591a3b53",
"md5": "d64ea710ff5b5e0ebc622a18735ffc2b",
"sha256": "be1d277febf2d507d06349fd5455c56a11d9177906050969b41802795c991ba8"
},
"downloads": -1,
"filename": "cutensor_cu11-2.0.1-py3-none-win_amd64.whl",
"has_sig": false,
"md5_digest": "d64ea710ff5b5e0ebc622a18735ffc2b",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": null,
"size": 137898114,
"upload_time": "2024-02-08T20:08:31",
"upload_time_iso_8601": "2024-02-08T20:08:31.924010Z",
"url": "https://files.pythonhosted.org/packages/42/49/ebeeb12e73eab21aff1249922a32e4153aae5bcccfab718ef463591a3b53/cutensor_cu11-2.0.1-py3-none-win_amd64.whl",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-02-08 20:06:20",
"github": false,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"lcname": "cutensor-cu11"
}
JSON