Skip to main content

Hyper optimized contraction trees for large tensor networks and einsums.

Project description

cotengra

tests codecov Codacy Badge Docs PyPI Anaconda-Server Badge

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:

  • drop-in einsum replacement
  • 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 cotengra, even if they don't provide einsum or tensordot but do have (batch) matrix multiplication

cotengra

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cotengra-0.6.0.tar.gz (3.3 MB view details)

Uploaded Source

Built Distribution

cotengra-0.6.0-py3-none-any.whl (171.8 kB view details)

Uploaded Python 3

File details

Details for the file cotengra-0.6.0.tar.gz.

File metadata

  • Download URL: cotengra-0.6.0.tar.gz
  • Upload date:
  • Size: 3.3 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.0.0 CPython/3.12.3

File hashes

Hashes for cotengra-0.6.0.tar.gz
Algorithm Hash digest
SHA256 dbd2d682d5eb40177246e8646dbabf0235650ca11f4986967b6ea9ee63549273
MD5 934298558a9565ac8b25f719e23682f4
BLAKE2b-256 45ce9239a7efe1c8afe35411551a403f6f6af8685a25dc9cb8b66475757f1582

See more details on using hashes here.

File details

Details for the file cotengra-0.6.0-py3-none-any.whl.

File metadata

  • Download URL: cotengra-0.6.0-py3-none-any.whl
  • Upload date:
  • Size: 171.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.0.0 CPython/3.12.3

File hashes

Hashes for cotengra-0.6.0-py3-none-any.whl
Algorithm Hash digest
SHA256 39b564cecc71b4b54b555854aacb06c32c74555dbcfb4c102634130709b043a0
MD5 d0ab0e29ca54cc2022ada1b659a06c71
BLAKE2b-256 99429548f9c1e2f2188caddd2cb5f1c9614d16b0036ec77900a03108a288acf4

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page