Powered by OpenAIRE graph
Found an issue? Give us feedback
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/ ZENODOarrow_drop_down
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/
ZENODO
Research
Data sources: ZENODO
Zenodo
Claim

JAX Quantum Research Suite: A Unified, Hardware-Accelerated, Differentiable Simulator for NISQ-Era Algorithm Research Across GPU and Cloud TPU Clusters

descriptionPublicationkeyboard_double_arrow_right Research Under curation English Publisher:Zenodo
Authors: Singh, Ashitesh;

doi: 10.5281/zenodo.20701873

JAX Quantum Research Suite: A Unified, Hardware-Accelerated, Differentiable Simulator for NISQ-Era Algorithm Research Across GPU and Cloud TPU Clusters

Abstract

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.

Powered by OpenAIRE graph
Found an issue? Give us feedback