message: [JAX] Bump xla_client._version and gate _ALLOW_ARG_PRUNING on the new version.
diff: @@ -50,7 +50,7 @@ profiler = _xla.profiler
  # Just an internal arbitrary increasing number to help with backward-compatible
  # changes.
  -_version = 19
  +_version = 20
  xla_platform_names = {
  'cpu': 'Host',
  