Documentation
Welcome to the jaxquantum documentation.
jaxquantum is a JAX-native quantum simulation toolkit for superconducting circuit devices and bosonic qubits. It serves as a drop-in QuTiP replacement while adding GPU/TPU acceleration, jax.vmap for parameter sweeps, jax.grad for differentiable physics, and jax.jit for compiled simulation loops.
Getting Started
New to jaxquantum? Start here:
- Installation — install via pip or from source
- Qarray — the core quantum array type: kets, operators, batch dimensions
- The Sharp Bits — common JAX pitfalls to avoid
Tutorials
Hands-on notebooks spanning the library:
| Tutorial | What you'll learn |
|---|---|
| Devices & Systems | Transmon spectroscopy, flux sweeps with vmap, parameter fitting with grad |
| Bosonic Codes | Cat, GKP, and Binomial qubit encodings; phase-space visualization; logical gates |
| Circuits | Gate-based circuits; unitary, Hamiltonian, and Kraus simulation modes; circuit optimization |
| Sparse Backends | SparseDIA and BCOO formats; when and how to use sparse operators; performance comparison |
Code Reference
Auto-generated API documentation for all modules: Reference
Energy Number Restricted (ENR) basis — documentation coming soon.