Adds colab_tpu_jax_version.ipynb
This Colab contains some useful commands to make sure that TPU runtime JAX version is not updated inadvertently, and provides some commands to figure out why which packages are compatible with which JAX versions.
See https://github.com/google/flax/discussions/2995
PiperOrigin-RevId: 521382578
