High-performance quantum systems simulation with JAX.

Get started Go to GitHub


Our quantum solvers are fully differentiable for use in optimal control, parameter estimation, experiment fitting, etc.

Computing gradients

GPU Accelerated

Transition seamlessly between CPU, GPU and TPU simulations, and run batches of simulations at once.

GPU support

Built on JAX

Benefit from the JAX ecosystem, built for both high-performance numerical computing and machine learning.

JAX documentation