Quantum Computing On JAX-TPU