Description
GPU-accelerated tensor linear algebra library for tensor contraction, reduction, and elementwise operations.
NVIDIA cuTENSOR is a GPU-accelerated tensor linear algebra library for tensor contraction, reduction, and elementwise operations. Using cuTENSOR, applications can harness the specialized tensor cores on NVIDIA GPUs for high-performance tensor computations and accelerate deep learning training and inference, computer vision, quantum chemistry, and computational physics workloads. By downloading and using this package you accept the terms and conditions of the associated license(s).