Fix c45d9db ("Drop Python 2 support from jax BUILD rule. #1965")
Fix c45d9db ("Drop Python 2 support from jax BUILD rule. #1965")

google/jax,google/jax,google/jax,google/jax,tensorflow/probability,tensorflow/probability