tensor-contraction 开源项目推荐（23 个）

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++

QuantumKitHub /TensorOperations.jl

Julia package for tensor contractions and related operations

ROCm /Tensile

[DEPRECATED] Moved to ROCm/rocm-libraries repo

jcmgray /cotengra

Hyper optimized contraction trees for large tensor networks and einsums

spacetimeengineer /spacetimeengine

一个Python实用程序，用于分析爱因斯坦场方程的给定解。基于症状。

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)

jcmgray /cotengrust

Rust accelerated contraction ordering primitives for tensor networks and einsums

bassoy /ttv

C++ Header-Only Library for High-Performance Tensor-Vector Multiplication

sdiehl /butler-portugal

Butler-Portugal algorithm for tensor canonicalization

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

jcmgray /einsum_bmm

einsum via batch matrix multiply

QuantumKitHub /TensorOperationsTBLIS.jl

Julia wrapper for TBLIS with TensorOperations.jl

MatthiasReumann /t1m

A header-only library for transposition-free tensor contractions

kaushikcfd /feinsum

Database approach for generating near-roofline Einstein Summation kernels.

jcmgray /einbench

benchmark and verification pairwise tensor contractions for einsum implementations