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

Uploaded Source

Built Distribution

cotengra-0.5.4-py3-none-any.whl (147.2 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: cotengra-0.5.4.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.4.tar.gz
Algorithm Hash digest
SHA256 56e290c9ca25d59a72cc6d6cc7915aed21f6f50002acff2edf38cc94c0de1ae6
MD5 91d439fde3445d1a83e037e24717e451
BLAKE2b-256 b65bfbdb773056aa1dfb3e472e73f537f14d4562f09845774d76288767fe96ee

See more details on using hashes here.

File details

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

File metadata

  • Download URL: cotengra-0.5.4-py3-none-any.whl
  • Upload date:
  • Size: 147.2 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.4-py3-none-any.whl
Algorithm Hash digest
SHA256 6abd6e7f8b7258e368653daacdbcb2676405aa215e8bb34b8ad56b4226c0ad85
MD5 59e82e40e1b911e1c3037f21fc0a5451
BLAKE2b-256 860ac464eacdb0f9135009b517e07690d7629a8dd3d5b3e8043bc2acac8f6cff

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