Blocks installation of jax 0.2.17 which is incompatible with current example.
Blocks installation of jax 0.2.17 which is incompatible with current example.

PiperOrigin-RevId: 384163140

tensorflow/tfx,tensorflow/tfx