Skip to main content

JAX backend for QuTiP

Project description

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

Unitary Fund Powered by NumFOCUS

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.

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

qutip_jax-0.1.1.tar.gz (33.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

qutip_jax-0.1.1-py3-none-any.whl (28.0 kB view details)

Uploaded Python 3

File details

Details for the file qutip_jax-0.1.1.tar.gz.

File metadata

  • Download URL: qutip_jax-0.1.1.tar.gz
  • Upload date:
  • Size: 33.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for qutip_jax-0.1.1.tar.gz
Algorithm Hash digest
SHA256 9177fdd58dd7320166f326ad74395c7883412f459a6cc3a470d4d65271432876
MD5 71f0bca93bc321e0e76ef7bba2e68d09
BLAKE2b-256 c691837887d247addb171d45f4a9ec56721bb14e84f378609fb65076457f052c

See more details on using hashes here.

Provenance

The following attestation bundles were made for qutip_jax-0.1.1.tar.gz:

Publisher: build.yml on qutip/qutip-jax

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file qutip_jax-0.1.1-py3-none-any.whl.

File metadata

  • Download URL: qutip_jax-0.1.1-py3-none-any.whl
  • Upload date:
  • Size: 28.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.12.9

File hashes

Hashes for qutip_jax-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 5db62504193298ebc197c03a98cdeaccf952811631de62b34546817594fbd6d4
MD5 3064ea8f48a2f0d8fa51f9361812c552
BLAKE2b-256 421a487656f04bc5ae581047aa3c08dba24e42346f2f094ddd12de22211ef323

See more details on using hashes here.

Provenance

The following attestation bundles were made for qutip_jax-0.1.1-py3-none-any.whl:

Publisher: build.yml on qutip/qutip-jax

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page