Skip to main content

Generic conjugate gradient solver that works with your favorite backend

Project description

PyConGrad

A somewhat optimized generic batch conjugate gradient algorithm that works with PyTorch, NumPy, CuPy, and whatever other backend you like as long as you code it up.

Heavily inspired by sbarratt/torch_cg and uses similar function signatures.

For more information, documentation, and detailed instructions, see the examples folder. In particular, notebook 2 in that folder discusses how to exactly duplicate the behavior of the original torch_cg.

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

congrad-0.0.1.tar.gz (8.5 kB view details)

Uploaded Source

Built Distribution

congrad-0.0.1-py3-none-any.whl (9.8 kB view details)

Uploaded Python 3

File details

Details for the file congrad-0.0.1.tar.gz.

File metadata

  • Download URL: congrad-0.0.1.tar.gz
  • Upload date:
  • Size: 8.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.0 CPython/3.12.5

File hashes

Hashes for congrad-0.0.1.tar.gz
Algorithm Hash digest
SHA256 cc777dae6c83aa71a57aedbb3a0dc3124b1cfcefa960fd31333f7a57452bfca1
MD5 2518f276d2dff6e8574a65f09a466d84
BLAKE2b-256 5047750313b4fda8ae05970331a4e489c0f62d849a632d6562b151c61e274e24

See more details on using hashes here.

File details

Details for the file congrad-0.0.1-py3-none-any.whl.

File metadata

  • Download URL: congrad-0.0.1-py3-none-any.whl
  • Upload date:
  • Size: 9.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.0 CPython/3.12.5

File hashes

Hashes for congrad-0.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 c6e17060ef08aa431a02b58b753351b63b45a97ecdc1a2082467bd1ec744f16d
MD5 55e517369a606d9c03dc2a0751b24db5
BLAKE2b-256 55e0526c441287c7035cc3f63cc098bf46af726c6ca1ce36d0d3323f33f6b65a

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 Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page