Computing magic with Neural Quantum States

Accompanying code for the paper "Non-stabilizerness of Neural Quantum States" (https://arxiv.org/abs/2502.09725). The code is based on jax, flax and NetKet.

Content of the repository

• nqsmagic: folder containing the following subfolders:
  • benchmarks: contains the benchmarks from perfect sampling with MPS (see https://doi.org/10.1103/PhysRevLett.131.180401).
  • models: contains the Neural Quantum States (NQS) models.
  • operator: contains the NetKet operator for computing the Stabilizer Rényi Entropy (SRE) M2 with NQS.
  • params: contains the NQS variational parameters to represent the ground states of the Transverse-Field Ising model and the J1-J2 Heisenberg model.
  • utils: contains some utility functions.
• examples: folder containing some examples of application.
• test: folder containing tests for the M2 stochastic estimators.

Installation

This package can be install directly from GitHub as:

git clone https://github.com/cqsl/nqsmagic.git
pip install -e ./nqsmagic

How to cite

If you use nqsmagic in your work, please consider citing it as:

@software{nqsmagic,
    author = {Sinibaldi, Alessandro},
    title = {nqsmagic package},
    url = {https://github.com/cqsl/nqsmagic},
    doi = {10.5281/zenodo.16941784},
    version = {0.0.1},
    year = {2025}
    }