jax

5599f473 - Revert to requiring Bazel 0.24 for JAX builds.

Revert to requiring Bazel 0.24 for JAX builds. It turns out TF does not fully support 0.25 yet. Add a check for a maximum bazel version to the build script. Fixes #702.