Make Jax buildable with the latest Bazel
Make Jax buildable with the latest Bazel

Without this change build script fails with the following message: 
```
Encountered error while reading extension file 'tensorflow/workspace.bzl': no such package '@org_tensorflow//tensorflow': The native http_archive rule is deprecated. load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive") for a drop-in replacement.
```

This PR simply implements the suggestion in the error. I was able to successfully build after applying the change.
google/jax,google/jax,google/jax,google/jax,tensorflow/probability,tensorflow/probability