tensor-contraction
There are 18 repositories under tensor-contraction topic.
dgasmith/opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
romeric/Fastor
A lightweight high performance tensor algebra framework for modern C++
Jutho/TensorOperations.jl
Julia package for tensor contractions and related operations
ROCm/Tensile
Stretching GPU performance for GEMMs and tensor contractions.
jcmgray/cotengra
Hyper optimized contraction trees for large tensor networks and einsums
spacetimeengineer/spacetimeengine
A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.
tensorly/torch
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
Einsums/Einsums
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
springer13/tcl
Tensor Contraction C++ Library
ROCm/MISA
Machine Intelligence Shader Autogen. AMDGPU ML shader code generator. (previously iGEMMgen)
bassoy/ttv
C++ Header-Only Library for High-Performance Tensor-Vector Multiplication
jcmgray/cotengrust
Rust accelerated contraction ordering primitives for tensor networks and einsums
jcmgray/einsum_bmm
einsum via batch matrix multiply
KennethBenicio/MSc-Multilinear-Algebra
Collection of basic and advanced Tensor Algebra operations using Matlab and Python.
flame/tblis-strassen
Strassen's Algorithm for Tensor Contraction
MatthiasReumann/t1m
Transposition-free Complex Tensor Contractions
QuantumKitHub/TensorOperationsTBLIS.jl
Julia wrapper for TBLIS with TensorOperations.jl
kaushikcfd/feinsum
Database approach for generating near-roofline Einstein Summation kernels.