Yanked

This release has been yanked by its maintainers, and will be ignored by installers, except when explicitly specified.
Consider using release 2.8.0 instead.
Reason given by maintainers: placeholder

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.

    • Tensor reductions (including partial reductions).

    • Element-wise tensor operations:

      • Support for various activation functions.

      • 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:

pip install cutensor-cuXX

where XX is the CUDA major version (currently CUDA 11 is supported). The package cutensor (without the -cuXX suffix) is considered deprecated.

Project links

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
  • Yanked:
Newer release available (2.8.0)

Owner

PyPI data
Data sourced directly from PyPI's database.

7 maintainers

PyPI data
Data sourced directly from PyPI's database.
Avatar for bkloss from gravatar.com bkloss Avatar for cpnv from gravatar.com cpnv Avatar for dlowell from gravatar.com dlowell
+4 more Avatar for JeremyWangNVDA from gravatar.com JeremyWangNVDA Avatar for mhohnerbach_nvidia from gravatar.com mhohnerbach_nvidia Avatar for mtjrider from gravatar.com mtjrider Avatar for shpface from gravatar.com shpface

Credits

Author: NVIDIA Corporation

License

NVIDIA Proprietary Software

Tags

cuda nvidia machine learning tensor network high-performance computing

Classifiers

Report project as malware

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

cutensor_cu12-0.0.1.dev0-py3-none-manylinux2014_x86_64.whl (1.9 kB view details)

Uploaded Python 3

File details

Details for the file cutensor_cu12-0.0.1.dev0-py3-none-manylinux2014_x86_64.whl.

File metadata

  • Download URL: cutensor_cu12-0.0.1.dev0-py3-none-manylinux2014_x86_64.whl
  • Upload date:
  • Size: 1.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.3.0 pkginfo/1.8.2 requests/2.26.0 setuptools/59.4.0 requests-toolbelt/0.9.1 tqdm/4.62.2 CPython/3.8.13

File hashes

Hashes for cutensor_cu12-0.0.1.dev0-py3-none-manylinux2014_x86_64.whl
Algorithm Hash digest
SHA256 e4ef8045224d3761a1299dbab07ff1adb27932ad74fc0b2b568eb45d9320741f
MD5 0eb5ee74dc304476633ca0efc35ed3a1
BLAKE2b-256 62b038f57e8676ba1394206d8be35d98f3d812ae506523009ee1b391e37a2628

See more details on using hashes here.

Release history Release notifications | RSS feed

2.8.0

3 files

2.7.0

3 files

2.6.0

3 files

2.5.0

3 files

2.4.1

3 files

2.3.1

3 files

2.3.0

3 files

2.2.0

3 files

2.0.2

3 files

2.0.1

3 files

2.0.0

3 files

1.7.0

3 files

1.6.2

2 files

This release
Yanked

0.0.1.dev0 This release

1 file

Yanked reason: placeholder

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page