AshiteshSingh/Tpu-Accelerated-Quantum-JAX ? reverse-engineered prompt

Build me a pure JAX quantum simulator that can run big state vector circuits fast on GPUs and Google Cloud TPUs.

I want it to let me simulate quantum gates, run things like Grover’s search and Shor’s algorithm, and support differentiable circuits so I can use gradients for quantum machine learning experiments. Please include a clean way to run benchmarks, a few example notebooks or scripts, and a simple test suite so I can verify it works. It should work on local machines too, but scale up when TPU or GPU hardware is available.

If you need to, look up the current JAX TPU and GPU setup docs online so the install and run steps are realistic. Also make sure the project feels research friendly, with clear examples and enough comments that I can follow along.