Explainax: An Explainable AI Library for JAX Installation | Documentation explainax is a JAX library for model explanation and interpretation. Installation Simply install via pypi: pip install explainax --upgrade Or upgrade to the latest version of explainax: pip install git+https://github.com/BirkhoffG/explainax.git --upgrade