Quantax documentation

Quantax is a Python package for flexible neural quantum states based on QuSpin, JAX, and Equinox.

• Installation

• API
  • global_defs
  • sites
  • symmetry
  • operator
  • nn
  • model
  • state
  • sampler
  • optimizer
  • utils