Handle the new JAX bind-params API

[gcarleo]

Summary

• normalize primitive.get_bind_params(...) in the interpreter
• gate the compatibility behavior on jax >= 0.9.2
• add regression tests covering both the legacy tuple-return API and the new dict-return API

Root cause

For newer JAX versions, some primitives such as integer_pow, lt, and jit return a plain dict from primitive.get_bind_params(...) instead of (subfuns, params). folx still unpacked the result unconditionally, which caused failures like ValueError: not enough values to unpack.

Validation

• uv run --no-dev --with pytest --with pytest-xdist pytest test/test_interpreter.py test/test_customjvp.py test/test_operator.py
• uv run --python 3.13 --no-dev --with pytest --with pytest-xdist --with jax==0.9.2 --with jaxlib==0.9.2 pytest test/test_interpreter.py test/test_customjvp.py test/test_operator.py

Both runs passed locally.

[gcarleo]

@microsoft-github-policy-service agree

[gcarleo]

Closing this in favor of a clean replacement PR based directly on microsoft/main, without the unrelated netket-only commits that caused conflicts.

Replacement: will link in the next comment once GitHub returns the new PR URL.

[gcarleo]

Replacement PR: #42

This one is based directly on microsoft/main and contains only the bind-params compatibility fix plus its regression test, so it avoids the unrelated fork-divergence conflicts.