message: [XLA:Python] Change import paths in xla_client to use relative paths to simplify jaxlib build rules.
  Currently we need to rewrite this path using sed as part of a jaxlib build, but if we make the path relative, we don't need the rewrite.
diff: @@ -28,6 +28,8 @@ import inspect
  import os
  from typing import List, Sequence, Tuple, Union
  +from . import xla_extension as _xla
  +
  from absl import logging
  import numpy as np
  @@ -36,8 +38,6 @@ import numpy as np
  # of TensorFlow. If we use protocol buffers here, then importing both jaxlib
  # and TensorFlow may fail with duplicate protocol buffer message definitions.
  -from tensorflow.compiler.xla.python import xla_extension as _xla
  -
  # Most functions are snake_case for consistency with other modules, some
  # method names are CamelCase for consistency with XLA.
  # pylint: disable=invalid-name
  