tf-explain is available on Pypi as an alpha release. To install it:

pip install tf-explain

Tensorflow Compatibility

tf-explain is compatible with Tensorflow 2. It is not declared as a dependency to let you choose between CPU and GPU versions. Additionally to the previous install, run:

pip install tensorflow==2.1.0