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

Uploaded Source

Built Distribution

cotengra-0.5.3-py3-none-any.whl (147.0 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: cotengra-0.5.3.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.3.tar.gz
Algorithm Hash digest
SHA256 1058ab4fe68b1ac2ff081ff8b890bbf150f79d2011bf55edde0d8b9c9231297e
MD5 acc3ce0d29041c683c136fa54cdf08cd
BLAKE2b-256 21244609225b705874a156d387c2c3685c5772780d5605d71fbf502ad88470a0

See more details on using hashes here.

File details

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

File metadata

  • Download URL: cotengra-0.5.3-py3-none-any.whl
  • Upload date:
  • Size: 147.0 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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 e1ee5a333c1ff664e95fe53a46bea5fa9aae8f40e9b837167010843844bedbfe
MD5 19b676109176a439183138da7512c3cc
BLAKE2b-256 484c1a271b37b1bc757b3908311e9df79f64ed300f50aab1b26eb101176743e8

See more details on using hashes here.

Supported by

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