JAX Quantum Research Suite enables differentiable simulation of quantum circuits using GPU and Cloud TPU clusters, suggesting advancements in quantum computing.
We present the JAX Quantum Research Suite, a high-performance, differentiable quantum circuit simulator spanning two co-existing hardware acceleration layers: a GPU division targeting NVIDIA RTX-class consumer GPUs via CUDA and a Cloud TPU division distributed across Google Cloud TPU v5e-16 and v6e-64 VM clusters. The suite is implemented in pure JAX (jax.numpy, jax.lax) for all statevector primitives, enabling seamless composition with jax.grad, jax.vmap, and jax.sharding.Positional Sharding.
No takes yet. Share an insight, caveat, or question.
Ashitesh Singh (2026) studied this question.
Synapse has enriched 5 closely related papers on similar clinical questions. Consider them for comparative context: