jax

b53b6347 - [jaxlib] Generate .pyi files for the _jax extension via nanobind's stubgen