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

Uploaded Source

Built Distribution

cotengra-0.5.6-py3-none-any.whl (148.0 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for cotengra-0.5.6.tar.gz
Algorithm Hash digest
SHA256 4e8ab61ea569eced25b602ca72aae0854f8825da76190664fe9c1414a55499cb
MD5 fcffd8325348a5ef9c93b1e97734f8d0
BLAKE2b-256 9250527fc5a893fb63b126401139e1e12f63a6ebd2a85bae60cfcf838af62c8e

See more details on using hashes here.

File details

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

File metadata

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

File hashes

Hashes for cotengra-0.5.6-py3-none-any.whl
Algorithm Hash digest
SHA256 d2e1fa9f39e80e9e02fd750112d6722cc22538ce9a60d3aa247143f3db0e59f3
MD5 3d3a83dadae43c4ce8712ffa6130c5ba
BLAKE2b-256 13d818a2656d09aa813e46f3969c796c131e883bc15d8435a408b51801e3417f

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