⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Jun 14, 2025 - Python
#
tensor-contraction
Here are 8 public repositories matching this topic...
[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
Nov 6, 2025 - Python
Hyper optimized contraction trees for large tensor networks and einsums
-
Updated
Nov 3, 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
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
-
Updated
Jul 30, 2025 - Python
einsum via batch matrix multiply
-
Updated
Nov 29, 2023 - Python
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
Nov 4, 2025 - Python
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."