einsum

There are 2 repositories under einsum topic.

dgasmith / opt_einsum
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
contraction einsum gpu-acceleration performance python tensor tensor-contraction
Language:Python 805
mcabbott / Tullio.jl
⅀
automatic-differentiation einsum julia notation tensor
Language:Julia 585
cotengra
jcmgray / cotengra
Hyper optimized contraction trees for large tensor networks and einsums
tensor-networks tensor-contraction tensor-network quimb opt-einsum einsum tensor
Language:Python 167
under-Peter / OMEinsum.jl
One More Einsum for Julia! With runtime order-specification and high-level adjoints for AD
automatic-differentiation contraction einsum
Language:Julia 165
mcabbott / TensorCast.jl
It slices, it dices, it splices!
tensor notation julia macros einsum broadcasting slice
Language:Julia 135
arviz-devs / xarray-einstats
Stats, linear algebra and einops for xarray
python numpy xarray scipy scipy-stats xarray-extension einsum linear-algebra
Language:Python 58
bdusell / semiring-einsum
Generic PyTorch implementation of einsum that supports different semirings
einsum pytorch semirings
Language:Python 42
Einsums / Einsums
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
einsum cpp tensor dense-matrices linear-algebra matrix matrix-computations matrix-library cp-decomposition tensor-decomposition tucker-decomposition tensor-contraction scientific-computing tensors cpp20
Language:C++ 34
pseewald / fortran-einsum-example
A generic implementation of tensor einsum in Fortran.
tensor einsum generic-programming fortran
Language:Fortran 24
braun-steven / simple-einet
An implementation of EinsumNetworks in PyTorch.
machine-learning sum-product-networks einsum deep-learning
Language:Jupyter Notebook 18
Linux-cpp-lisp / opt_einsum_fx
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
einsum optimization pytorch pytorch-fx
Language:Python 18
hhaoyan / opt-einsum-torch
Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
einsum gpu-computing pytorch scientific-computing
Language:Python 14
f-dangel / einconv
Convolutions and more as einsum for PyTorch
convolution einsum einsum-notation fisher ggn im2col jvp kfac ntk pytorch tensornetwork transpose-convolution vjp
Language:Python 12
jcmgray / cotengrust
Rust accelerated contraction ordering primitives for tensor networks and einsums
contraction-order einsum tensor-networks contraction tensor tensor-contraction tensor-network
Language:Rust 11
jcmgray / einsum_bmm
einsum via batch matrix multiply
einsum tensor tensor-contraction tensor-networks
Language:Python 10
Syngular
antoine311200 / Syngular
A Tensor Network package for Machine Learning and Quantum Computing in Python.
algebra einsum matrix-product-states quantum-computing renormalization-group tensor tensor-network tensor-train matrix-product-operators
Language:Python 9
EinExprs.jl
bsc-quantic / EinExprs.jl
Einsum Expressions in Julia
contraction einsum julia tensor tensor-network symbolic-expressions
Language:TeX 7
FPSG-UIUC / teaal-compiler
einsum hifiber
Language:Python 6
ShawnZhong / CS759-Spring-2020-Final-Project
Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
einsum einsum-notation pytorch performance performance-optimization benchmarking profiling cutensor dynamic-programming
Language:Python 6
stoianmihail / Netzwerk
A collection of state-of-the-art contraction ordering algorithms. https://arxiv.org/abs/2209.12332
contraction-order cotengra dynamic-programming einsum join-ordering opt-einsum optimal-algorithm tensornetwork
Language:Python 5
bsc-quantic / OptimizedEinsum.jl
Julia rewrite of opt_einsum Python package
julia tensor-networks contraction-order einsum
Language:Julia 3
vietnamican / einsum_practice
All you need is einsum
einsum
Language:Jupyter Notebook 3
braun-steven / EinsumNetworks
Personal fork of the official EinsumNetworks implementation with a few enhancements.
einsum machine-learning sum-product-networks
Language:Python 2
lgeiger / mypy-einsum
MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations
deep-learning einsum jax mypy mypy-plugins numpy pytorch
Language:Python 2
adtzlr / named-einsum
NumPy's Einsum, but with named subscripts.
einsum numpy python
Language:Python 1
alexancp / einsumpath-to-eT
einsum
Language:Python 1
bsc-quantic / Tensors.jl
Tensors and Einstein Summation in Julia
einsum tensor
Language:Julia
hzhangxyz / easy-ast
ast einstein-notation einsum python
Language:Python
toddsundsted / einsum
Unoptimized, pure-Ruby implementation of a subset of Numpy einsum.
einsum
Language:Ruby

einsum

dgasmith / opt_einsum

mcabbott / Tullio.jl

jcmgray / cotengra

under-Peter / OMEinsum.jl

mcabbott / TensorCast.jl

arviz-devs / xarray-einstats

bdusell / semiring-einsum

Einsums / Einsums

pseewald / fortran-einsum-example

braun-steven / simple-einet

Linux-cpp-lisp / opt_einsum_fx

hhaoyan / opt-einsum-torch

f-dangel / einconv

jcmgray / cotengrust

jcmgray / einsum_bmm

antoine311200 / Syngular

bsc-quantic / EinExprs.jl

FPSG-UIUC / teaal-compiler

ShawnZhong / CS759-Spring-2020-Final-Project

stoianmihail / Netzwerk

bsc-quantic / OptimizedEinsum.jl

vietnamican / einsum_practice

braun-steven / EinsumNetworks

lgeiger / mypy-einsum

adtzlr / named-einsum

alexancp / einsumpath-to-eT

bsc-quantic / Tensors.jl

hzhangxyz / easy-ast

toddsundsted / einsum