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
Release history Release notifications | RSS feed
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)
Built Distribution
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | cc777dae6c83aa71a57aedbb3a0dc3124b1cfcefa960fd31333f7a57452bfca1 |
|
MD5 | 2518f276d2dff6e8574a65f09a466d84 |
|
BLAKE2b-256 | 5047750313b4fda8ae05970331a4e489c0f62d849a632d6562b151c61e274e24 |
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | c6e17060ef08aa431a02b58b753351b63b45a97ecdc1a2082467bd1ec744f16d |
|
MD5 | 55e517369a606d9c03dc2a0751b24db5 |
|
BLAKE2b-256 | 55e0526c441287c7035cc3f63cc098bf46af726c6ca1ce36d0d3323f33f6b65a |