netket.errors.HilbertIndexingDuringTracingError

exception netket.errors.HilbertIndexingDuringTracingError

Illegal attempt to use state indexing function from inside of a Jax function transformation.

This happens when you call functions such as states_to_numbers() or its opposite numbers_to_states() from within a jax.jit(), jax.grad(), jax.vjp() or similar jax function transformations.

There is currently no workaround rather than returning the arrays from the jax function and performing the conversion outside of it.