Skip to content

Hyper optimized contraction trees for large tensor networks and einsums - https://cotengra.readthedocs.io

License

Notifications You must be signed in to change notification settings

stoianmihail/cotengra

 
 

Repository files navigation

cotengra

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:

  • an explicit contraction tree object that can be flexibly built, modified and visualized
  • a 'hyper optimizer' that samples trees while tuning the generating meta-paremeters
  • dynamic slicing for massive memory savings and parallelism
  • support for hyper edge tensor networks and thus arbitrary einsum equations
  • paths that can be supplied to numpy.einsum, opt_einsum, quimb among others
  • performing contractions with tensors from many libraries via autoray

cotengra

About

Hyper optimized contraction trees for large tensor networks and einsums - https://cotengra.readthedocs.io

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 94.7%
  • Rust 5.3%