Throw informative errors if one uses operators/hilbert inside of jax (#1509)
Throws informative errors when attempting to use `hilbert` and `operators` objects inside of jax function transformations, suggesting workarounds. Lies the foundation for a systematic error system in netket.
Showing
- docs/api/api.md 10 additions, 10 deletionsdocs/api/api.md
- docs/api/errors.md 21 additions, 0 deletionsdocs/api/errors.md
- netket/__init__.py 3 additions, 0 deletionsnetket/__init__.py
- netket/errors.py 280 additions, 0 deletionsnetket/errors.py
- netket/experimental/operator/_fermion_operator_2nd.py 9 additions, 1 deletionnetket/experimental/operator/_fermion_operator_2nd.py
- netket/hilbert/discrete_hilbert.py 10 additions, 0 deletionsnetket/hilbert/discrete_hilbert.py
- netket/hilbert/homogeneous.py 13 additions, 1 deletionnetket/hilbert/homogeneous.py
- netket/operator/_bose_hubbard.py 9 additions, 1 deletionnetket/operator/_bose_hubbard.py
- netket/operator/_ising/numba.py 8 additions, 2 deletionsnetket/operator/_ising/numba.py
- netket/operator/_local_operator.py 17 additions, 1 deletionnetket/operator/_local_operator.py
- netket/operator/_pauli_strings.py 9 additions, 1 deletionnetket/operator/_pauli_strings.py
- netket/utils/static_number.py 1 addition, 1 deletionnetket/utils/static_number.py
- test/hilbert/test_hilbert.py 19 additions, 0 deletionstest/hilbert/test_hilbert.py
- test/models/test_fullspace.py 1 addition, 1 deletiontest/models/test_fullspace.py
- test/operator/test_operator.py 31 additions, 4 deletionstest/operator/test_operator.py
- test/utils/test_staticzero.py 56 additions, 0 deletionstest/utils/test_staticzero.py
Loading
Please register or sign in to comment