text: Title: Trax Import Error
  Body: ### Description
  Hello, I am having issues with the tensorflow imports while using trax, I've never experienced the issue before and not sure how to solve it.
  
  ### Environment information
  ---------------------------------------------------------------------------
  NotFoundError                             Traceback (most recent call last)
  <ipython-input-2-8b607b832815> in <module>
       11 from nltk.tokenize import sent_tokenize, word_tokenize
       12 from nltk.stem import WordNetLemmatizer
  ---> 13 import trax
       14 import trax.fastmath.numpy as np
       15 from trax import layers as tl
  
  ~/opt/anaconda3/lib/python3.7/site-packages/trax/__init__.py in <module>
       16 """Trax top level import."""
       17 
  ---> 18 from trax import data
       19 from trax import fastmath
       20 from trax import layers
  
  ~/opt/anaconda3/lib/python3.7/site-packages/trax/data/__init__.py in <module>
       67 from trax.data.inputs import UniformlySeek
       68 
  ---> 69 from trax.data.tf_inputs import add_eos_to_output_features
       70 from trax.data.tf_inputs import BertGlueEvalStream
       71 from trax.data.tf_inputs import BertGlueTrainStream
  
  ~/opt/anaconda3/lib/python3.7/site-packages/trax/data/tf_inputs.py in <module>
       30 import numpy as np
       31 import scipy
  ---> 32 import t5.data
       33 import tensorflow as tf
       34 import tensorflow_datasets as tfds
  
  ~/opt/anaconda3/lib/python3.7/site-packages/t5/__init__.py in <module>
       15 """Import API modules."""
       16 
  ---> 17 import t5.data
       18 import t5.evaluation
       19 import t5.models
  
  ~/opt/anaconda3/lib/python3.7/site-packages/t5/data/__init__.py in <module>
       15 """Import data modules."""
       16 
  ---> 17 from t5.data.dataset_providers import *  # pylint:disable=wildcard-import
       18 from t5.data.feature_converters import *  # pylint:disable=wildcard-import
       19 from t5.data.glue_utils import *  # pylint:disable=wildcard-import
  
  ~/opt/anaconda3/lib/python3.7/site-packages/t5/data/dataset_providers.py in <module>
       30 import dataclasses
       31 import gin
  ---> 32 from t5.data import utils
       33 from t5.data.preprocessors import tokenize as tokenize_preprocessor
       34 from t5.data.vocabularies import Vocabulary
  
  ~/opt/anaconda3/lib/python3.7/site-packages/t5/data/utils.py in <module>
       25 import gin
       26 import numpy as np
  ---> 27 from t5.data import vocabularies
       28 import tensorflow.compat.v2 as tf
       29 import tensorflow_datasets as tfds
  
  ~/opt/anaconda3/lib/python3.7/site-packages/t5/data/vocabularies.py in <module>
       21 
       22 import tensorflow.compat.v2 as tf
  ---> 23 import tensorflow_text as tf_text
       24 
       25 import sentencepiece as sentencepiece_processor
  
  ~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/__init__.py in <module>
       19 # pylint: disable=wildcard-import
       20 from tensorflow_text.python import keras
  ---> 21 from tensorflow_text.python import metrics
       22 from tensorflow_text.python.ops import *
       23 
  
  ~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/__init__.py in <module>
       18 
       19 # pylint: disable=wildcard-import
  ---> 20 from tensorflow_text.python.metrics.text_similarity_metric_ops import *
       21 
       22 # Public symbols in the "tensorflow_text.metrics" package.
  
  ~/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/text_similarity_metric_ops.py in <module>
       26 from tensorflow.python.framework import load_library
       27 from tensorflow.python.platform import resource_loader
  ---> 28 gen_text_similarity_metric_ops = load_library.load_op_library(resource_loader.get_path_to_datafile('_text_similarity_metric_ops.dylib'))
       29 
       30 
  
  ~/opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/framework/load_library.py in load_op_library(library_filename)
       56     RuntimeError: when unable to load the library or get the python wrappers.
       57   """
  ---> 58   lib_handle = py_tf.TF_LoadLibrary(library_filename)
       59   try:
       60     wrappers = _pywrap_python_op_gen.GetPythonWrappers(
  
  NotFoundError: dlopen(/Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib, 6): Symbol not found: __ZN10tensorflow15TensorShapeBaseINS_11TensorShapeEEC2EN4absl14lts_2020_02_254SpanIKxEE
    Referenced from: /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib
    Expected in: /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow/python/../libtensorflow_framework.2.dylib
   in /Users/lexroverts/opt/anaconda3/lib/python3.7/site-packages/tensorflow_text/python/metrics/_text_similarity_metric_ops.dylib
  ```
  OS: 10.15.7
  
  $ pip freeze | grep trax
  # your output here
  trax==1.3.9
  $ pip freeze | grep tensor
  # your output here
  mesh-tensorflow==0.1.18
  tensor2tensor==1.15.7
  tensorboard==2.5.0
  tensorboard-data-server==0.6.1
  tensorboard-plugin-wit==1.8.0
  tensorflow==2.5.0
  tensorflow-addons==0.12.1
  tensorflow-datasets==4.2.0
  tensorflow-estimator==2.5.0
  tensorflow-gan==2.0.0
  tensorflow-hub==0.11.0
  tensorflow-metadata==0.27.0
  tensorflow-probability==0.7.0
  tensorflow-text==2.4.3
  $ pip freeze | grep jax
  # your output here
  jax==0.2.9
  jaxlib==0.1.59
  $ python -V
  # Python 3.7.6
  ```
  
  
  