Quantax documentation

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

• Installation
  • Step 1 - Install JAX
  • Step 2 - Install Quantax
  • Supported platforms
• API
  • global_defs
  • sites
  • symmetry
  • operator
  • nn
  • model
  • state
  • sampler
  • optimizer