quex.backends.jax_sim

quex.backends.jax_sim#

Classes#

JaxSimulator

GPU/TPU-accelerated statevector simulator using Google JAX.

Module Contents#

class quex.backends.jax_sim.JaxSimulator(**kwargs)[source]#

Bases: quex.backends.numpy_sim.NumpySimulator

GPU/TPU-accelerated statevector simulator using Google JAX. Provides automatic differentiation capabilities.

get_state(host: bool = True)[source]#

Safely returns the state to standard Numpy if requested.