Add support nogil mode in JAX for Python 3.13

[vfdev-5]

The idea is to add a support of nogil mode in JAX for Python 3.13. According to Jake and Peter, this would likely involve updates to both jaxlib and to nanobind.

• [ ] Add a CI job to run jax tests on python 3.13.0rc1 with free threading (or nogil). Similarly to scikit-learn, numpy or scipy (e.g. https://github.com/scikit-learn/scikit-learn/pull/29191)

Context:

• https://py-free-threading.github.io/
• https://docs.python.org/3.13/howto/free-threading-extensions.html

cc @jakevdp, @hawkinsp

[wjakob]

Nanobind's master branch now supports free-threading. Here is more information about the feature: https://nanobind.readthedocs.io/en/latest/free_threaded.html

[hawkinsp]

Thanks @wjakob . We're loving nanobind, by the way!