jax

b2549761 - Avoid duplication of Bazel dependencies between //jax/_src/lib and //jaxlib.

Avoid duplication of Bazel dependencies between //jax/_src/lib and //jaxlib. Put all dependencies on //jaxlib and make //jax/_src/lib a pure forwarding rule. PiperOrigin-RevId: 758378774