Version 0.12.0 does not work with jax 4.11

[jnboehm]

The PR #1595 has been merged (e715881) into the main branch after the version bump to 0.12.0 (f981b29), hence there is a mismatch between the release notes and the files hosted on pypi.

steps to reproduce

1. download the tarball for version 0.12.0 from pypi (https://files.pythonhosted.org/packages/3b/0a/9f12ba5303219c12b68081c3483d155fcdf5318d42cc44af64803c50c926/numpyro-0.12.0.tar.gz)
2. extract the source files with tar xvf numpyro-0.12.0.tar.gz
3. observe that in line 10 the import statement from jax.interpreters.xla import xla_call_p is still present

Solution

Release another version (I guess 0.12.1) after the PR has been merged.

Sorry for the bother, I was a bit confused tripping over this error just now and it took me a while to figure out what the issue was, although the solution is quite simple.

[fehiepsi]

Thanks for the ping! I just happened that jax team just released 4.11 after the above PR. Let me make a minor release.

[jnboehm]

Cool, yeah it was kind of a coincidence that the PR would become relevant so soon. Thanks for the quick response! :)