Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/usr/lib/python3.8/site-packages/tensorflow/__init__.py", line 101, in <module>
    from tensorflow_core import *
  File "/usr/lib/python3.8/site-packages/tensorflow_core/__init__.py", line 40, in <module>
    from tensorflow.python.tools import module_util as _module_util
  File "/usr/lib/python3.8/site-packages/tensorflow/__init__.py", line 50, in __getattr__
    module = self._load()
  File "/usr/lib/python3.8/site-packages/tensorflow/__init__.py", line 44, in _load
    module = _importlib.import_module(self.__name__)
  File "/usr/lib/python3.8/importlib/__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "/usr/lib/python3.8/site-packages/tensorflow_core/python/__init__.py", line 85, in <module>
    from tensorflow.python.ops.standard_ops import *
  File "/usr/lib/python3.8/site-packages/tensorflow_core/python/ops/standard_ops.py", line 48, in <module>
    from tensorflow.python.ops.special_math_ops import *
  File "/usr/lib/python3.8/site-packages/tensorflow_core/python/ops/special_math_ops.py", line 30, in <module>
    import opt_einsum
ModuleNotFoundError: No module named 'opt_einsum'