⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Jun 14, 2025 - Python
#
tensor-contraction
Here are 21 public repositories matching this topic...
A lightweight high performance tensor algebra framework for modern C++
-
Updated
Jul 8, 2025 - C++
Julia package for tensor contractions and related operations
tensor tensor-contraction tensor-transposition tensor-operations tensor-permutation tensor-trace einstein-summation index-notation
-
Updated
Sep 30, 2025 - Julia
[DEPRECATED] Moved to ROCm/rocm-libraries repo
python machine-learning amd gpu assembly opencl dnn matrix-multiplication neural-networks gpu-acceleration blas hip gpu-computing tensors tensor-contraction gemm radeon auto-tuning
-
Updated
Oct 14, 2025 - Python
Hyper optimized contraction trees for large tensor networks and einsums
-
Updated
Oct 9, 2025 - Python
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
computer-algebra spacetime gravity connection research-tool riemann sympy tensor-algebra tensor-contraction einstein science-research gravity-simulation riemannian-geometry ricci-curvature einstein-coefficients metric-theory einstein-field-equations spacetime-toolkit
-
Updated
Apr 17, 2025 - Python
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
deep-neural-networks deep-learning pytorch tensor tensor-contraction tensor-learning tensor-networks factorized-cnns tensor-convolution-networks tensor-regression deep-tensor-learning tensorized-networks factorized-convolution tensor-regression-layers factorized-networks
-
Updated
Jun 9, 2024 - Python
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
cpp matrix linear-algebra scientific-computing tensor tensors tensor-contraction tensor-decomposition matrix-library cp-decomposition matrix-computations cpp20 tucker-decomposition einsum dense-matrices
-
Updated
Oct 1, 2025 - C++
Tensor Contraction C++ Library
-
Updated
Aug 22, 2019 - C++
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
-
Updated
Jul 30, 2025 - Python
Rust accelerated contraction ordering primitives for tensor networks and einsums
-
Updated
Oct 6, 2025 - Rust
C++ Header-Only Library for High-Performance Tensor-Vector Multiplication
fast c-plus-plus high-performance arrays blas tensor tensor-contraction multidimensional tensor-library multilinear-algebra tensor-vector-multiplications tensor-times-vector tensor-vector-multiplication
-
Updated
Dec 16, 2024 - C++
Collection of basic and advanced Tensor Algebra operations using Matlab and Python.
python matlab mathematics least-squares als cpd tensor-algebra tensor-contraction folding unfolding alternating-least-squares tensor-train kronecker-product parafac tensor-product multilinear-algebra tensor-operations khatri-rao multilinear-svd
-
Updated
Jul 14, 2022 - Jupyter Notebook
Strassen's Algorithm for Tensor Contraction
-
Updated
Jul 7, 2017 - C
einsum via batch matrix multiply
-
Updated
Nov 29, 2023 - Python
Implementation of Butler-Portugal algorithm for tensor canonicalization in Rust
-
Updated
Aug 29, 2025 - Rust
Julia wrapper for TBLIS with TensorOperations.jl
-
Updated
Aug 12, 2025 - Julia
A header-only library for transposition-free tensor contractions
-
Updated
Jul 5, 2025 - C++
Database approach for generating near-roofline Einstein Summation kernels.
python sql gpu opencl linear-algebra code-generation tensors multidimensional-arrays tensor-contraction einstein-summation
-
Updated
Oct 17, 2025 - Python
benchmark and verification pairwise tensor contractions for einsum implementations
-
Updated
Sep 30, 2025 - Jupyter Notebook
Improve this page
Add a description, image, and links to the tensor-contraction topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the tensor-contraction topic, visit your repo's landing page and select "manage topics."