jax

0e05a798 - Split some submodules out of //jax under Bazel.

Split some submodules out of //jax under Bazel. Add separate BUILD targets * :version - for version.py * _src/lib - wrapping the jaxlib shims. * :util - for util.py * :config - for config.py PiperOrigin-RevId: 515307923