High-performance, differentiable quantum state-vector & tensor network simulator in 100% pure JAX (no classical framework overhead). Accelerated on NVIDIA GPUs and Google Cloud TPU v6e-64/v5e VM clusters up to 37 qubits! Supported by Google's TPU Research Cloud (TRC) program.
python machine-learning google-cloud quantum-computing nvidia-gpu tpu grovers-algorithm jax quantum-simulation shors-algorithm xla quantum-ml quantum-computing-research tpu-research-cloud tpu-accelerated-quantum-jax jax-quantum-research-suite jax-quantum nisq-algorithms
-
Updated
Jul 20, 2026 - Jupyter Notebook