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

Uploaded Source

Built Distribution

cotengra-0.5.5-py3-none-any.whl (147.6 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: cotengra-0.5.5.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.5.tar.gz
Algorithm Hash digest
SHA256 02248fb40477d17933413e991c311310c712528725f904189052232ca6013ff6
MD5 eaf76a1b0ff5fd4e2ff6bb6208dfddd2
BLAKE2b-256 4f7c9a579ad04f69ce59dbb6a53408dad24fda7feb2322e7c2b2a7cf19f48a5c

See more details on using hashes here.

File details

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

File metadata

  • Download URL: cotengra-0.5.5-py3-none-any.whl
  • Upload date:
  • Size: 147.6 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.5-py3-none-any.whl
Algorithm Hash digest
SHA256 2452e2485efa76346bd6bc1264df5eb91fc89f9d0acd11f661c91392b8d9c7b0
MD5 7761793a9bbf60bfbc104d4787889b2b
BLAKE2b-256 f268c60ff0d3a7b62c18ead5bc02b42e9ef3b11a626f51603b30f23c1521577d

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