qutip-jax: JAX backend for QuTiP
A plug-in for QuTiP providing a JAX linear-algebra backend.
Backing the linear algebra operations with JAX extends QuTiP's
capability to work with a GPU. Furthermore, it allows QuTiP's Qobj class to
benefit from auto differentiation.
We are actively developing the plug-in and in the pre-alpha state. Therefore it is a work-in-progress and is not ready for use yet. You are welcome to contribute or take part in the development by posting issues, PRs and through discussions.
Documentation
https://qutip-jax.readthedocs.io/en/latest/index.html
Support
We are proud to be affiliated with Unitary Fund and NumFOCUS. QuTiP development is supported by Nori's lab at RIKEN, by the University of Sherbrooke, and by Aberystwyth University, among other supporting organizations.
Contact
Contact Shahnawaz Ahmed (shahnawaz.ahmed95@gmail.com), Eric Giguère(eric.giguere@calculquebec.ca) or anyone from the QuTiP admin team (https://qutip.org/devs.html) for more information.
Release files for qutip-jax 0.1.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| qutip_jax-0.1.1.tar.gz | 33.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| qutip_jax-0.1.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 61.8 kB
Release files / qutip_jax-0.1.1.tar.gz
| Download URL | qutip_jax-0.1.1.tar.gz |
|---|---|
| Size | 33.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
9177fdd58dd7320166f326ad74395c7883412f459a6cc3a470d4d65271432876
|
|
BLAKE2b-256 checksum How to use checksums |
c691837887d247addb171d45f4a9ec56721bb14e84f378609fb65076457f052c
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.12.9
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on May 29, 2025.
Transparency logRelease files / qutip_jax-0.1.1-py3-none-any.whl
| Download URL | qutip_jax-0.1.1-py3-none-any.whl |
|---|---|
| Size | 28.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
5db62504193298ebc197c03a98cdeaccf952811631de62b34546817594fbd6d4
|
|
BLAKE2b-256 checksum How to use checksums |
421a487656f04bc5ae581047aa3c08dba24e42346f2f094ddd12de22211ef323
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.12.9
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on May 29, 2025.
Transparency log
