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.5.1.tar.gz (3.2 MB view details)

Uploaded Source

Built Distribution

cotengra-0.5.1-py3-none-any.whl (144.9 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: cotengra-0.5.1.tar.gz
  • Upload date:
  • Size: 3.2 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.6

File hashes

Hashes for cotengra-0.5.1.tar.gz
Algorithm Hash digest
SHA256 20855f87c1df6534dd1dad17302ad3a5dcc88a86e38a9ce93117b24545dc38cc
MD5 276137f9e55013dfa54778ff393abb08
BLAKE2b-256 a801e198c106683d9588dae3881c293bc22e36a342eb2adbe8e3bd0d6a2d679a

See more details on using hashes here.

File details

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

File metadata

  • Download URL: cotengra-0.5.1-py3-none-any.whl
  • Upload date:
  • Size: 144.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.11.6

File hashes

Hashes for cotengra-0.5.1-py3-none-any.whl
Algorithm Hash digest
SHA256 4b3ca54681d991a6bd4d2a265e47b8688e969b4c2c99ddd748164f0c724fe45e
MD5 5f3675c63ef2f0da3191bcf254855b3b
BLAKE2b-256 b2f22acb7557f2df65f64f2c4053551269e0ffc54f397422d2e354f5fed017c9

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