[Take 2] Implement Jax-compatible PauliString (#1510)
This is also a remanding of the original PR by @wdphy16 , following the
new infrastructure for Jax-compatible operators.
resuscitates #968
@wdphy16 as you are the original author of the code, could you please go
back through it and comment it throughout? It's very complicated and
it's very hard to follow what the generation of masks do...
---------
authored-by:
Clemens Giuliani <clemens@inailuig.it>
Showing
- CHANGELOG.md 1 addition, 0 deletionsCHANGELOG.md
- docs/api/errors.md 1 addition, 0 deletionsdocs/api/errors.md
- docs/api/operator.md 1 addition, 0 deletionsdocs/api/operator.md
- netket/errors.py 80 additions, 0 deletionsnetket/errors.py
- netket/experimental/operator/_fermion_operator_2nd.py 1 addition, 1 deletionnetket/experimental/operator/_fermion_operator_2nd.py
- netket/operator/__init__.py 2 additions, 1 deletionnetket/operator/__init__.py
- netket/operator/_discrete_operator_jax.py 44 additions, 6 deletionsnetket/operator/_discrete_operator_jax.py
- netket/operator/_ising/numba.py 1 addition, 1 deletionnetket/operator/_ising/numba.py
- netket/operator/_pauli_strings/__init__.py 17 additions, 0 deletionsnetket/operator/_pauli_strings/__init__.py
- netket/operator/_pauli_strings/base.py 150 additions, 314 deletionsnetket/operator/_pauli_strings/base.py
- netket/operator/_pauli_strings/jax.py 377 additions, 0 deletionsnetket/operator/_pauli_strings/jax.py
- netket/operator/_pauli_strings/numba.py 301 additions, 0 deletionsnetket/operator/_pauli_strings/numba.py
- netket/sampler/rules/hamiltonian.py 5 additions, 1 deletionnetket/sampler/rules/hamiltonian.py
- test/operator/test_hamiltonian.py 15 additions, 0 deletionstest/operator/test_hamiltonian.py
- test/operator/test_operator.py 5 additions, 0 deletionstest/operator/test_operator.py
Loading
Please register or sign in to comment