Skip to main content

Hyper optimized contraction trees for large tensor networks and einsums.

Project description

cotengra

tests codecov Codacy Badge 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.0.tar.gz (3.2 MB view details)

Uploaded Source

Built Distribution

cotengra-0.5.0-py3-none-any.whl (143.9 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for cotengra-0.5.0.tar.gz
Algorithm Hash digest
SHA256 ca60993bd67f41d1357a8ce5e9c8d84bd26157b7ce81db8a0a5d320a5696f7e1
MD5 784395b30f7c65cac016ecf3a8aa3352
BLAKE2b-256 cfdffd7e77a46d25d48e3d39adacfe06ace6ba8f5a6917dd4a0fafcf18fb6e08

See more details on using hashes here.

File details

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

File metadata

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

File hashes

Hashes for cotengra-0.5.0-py3-none-any.whl
Algorithm Hash digest
SHA256 c0af7cac638d9d7206a1a5631c32696384525b20fde1daa9364167b2a898b5e5
MD5 49bd26882b352443ce2af4d4895df6d8
BLAKE2b-256 6759f5bd998e67e34fe9da6a6f72cee0a43e7053c2e5ce40216ba78bf34e8bff

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