cutensor
Here are 5 public repositories matching this topic...
Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
-
Updated
Sep 25, 2020 - Python
Numeric Library for Deep Learning purposes
-
Updated
Apr 2, 2021 - C#
Tensor contractions for AWS Trainium via NKI (cuTENSOR-equivalent) — einsum with contraction planning, CP/PARAFAC and Tucker decompositions, density-fitted post-Hartree-Fock patterns.
-
Updated
Apr 21, 2026 - Python
Accelerate NumPy and SciPy operations by using NVIDIA CUDA GPUs with this multi-dimensional array library.
-
Updated
Jun 29, 2026
Improve this page
Add a description, image, and links to the cutensor topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the cutensor topic, visit your repo's landing page and select "manage topics."