Quick Install
The PyPI distribution is named torchxai-tools; the Python import name is torchxai.
pip install torchxai-tools
uv add torchxai-tools
poetry add torchxai-tools
After installation:
from torchxai.explainers import SaliencyExplainer # import name is torchxai
Optional extras
GPU-accelerated image examples require torchvision:
pip install "torchxai-tools[vision]"
uv add "torchxai-tools[vision]"
Install from source
To install the latest development version directly from GitHub:
pip install git+https://github.com/atriaml/torchxai.git
uv add git+https://github.com/atriaml/torchxai.git
git clone https://github.com/atriaml/torchxai.git
cd torchxai
pip install -e .