Support building with Bzlmod

[meteorcloudy]

This enables bazel build --config=bzlmod //jaxlib/tools:jaxlib_wheel

But still needs to work on a solution for hermetic python toolchains.