High-performance, GPU-accelerated and differentiable quantum systems simulation with JAX.
scientific-computing quantum-simulation differentiable-programming gpu-acceleration python jax