Hyper optimized contraction trees for large tensor networks and einsums
cotengra
is a python library for contracting tensor networks or einsum
expressions involving large numbers of tensors - the main docs can be found
at cotengra.readthedocs.io.
Some of the key feautures of cotengra
include:
einsum
replacementnumpy.einsum
, opt_einsum
, quimb
among otherscotengra
,
even if they don't provide einsum
or tensordot
but do have (batch) matrix
multiplication