GitHub repository: ChenAo-Phys/quantax
Primary language: Python
Description: Flexible neural quantum states based on QuSpin, JAX, and Equinox