torch>=2.1
torchvision>=0.16
numpy<2,>=1.22
scipy>=1.8
pandas>=1.5
matplotlib>=3.5
scikit-learn>=1.0
Pillow>=9.0
pydantic<3,>=2.0
PyYAML>=6.0
tqdm>=4.62
psutil>=5.9
requests>=2.28
huggingface-hub>=0.23
transformers>=4.36
diffusers>=0.28
timm>=0.9
torchmetrics>=1.0
tensorboard>=2.10
zennit>=0.4.6
blobfile>=2.0
wget>=3.2
seaborn>=0.11
torch-kmeans>=0.2

[clip]
open-clip-torch>=2.20

[datasets]
wilds>=2.0
kaggle>=1.5
kagglehub>=0.2

[full]
captum
corelay>=0.2.1
fastapi>=0.110
flask-cors>=3.0
flask>=2.2
h5py>=3.7
httpx>=0.27
kaggle>=1.5
kagglehub>=0.2
mpi4py>=3.1
omegaconf
omegaconf>=2.3
onnx2torch>=1.5
onnx>=1.15
onnxruntime>=1.17
open-clip-torch>=2.20
pygame>=2.1
python-multipart>=0.0.9
uvicorn[standard]>=0.27
virelay>=0.4.0
wilds>=2.0
zennit-crp>=0.6.0

[mpi]
mpi4py>=3.1

[onnx]
onnx>=1.15
onnxruntime>=1.17
onnx2torch>=1.5

[pathldm]
omegaconf

[pygame]
pygame>=2.1

[rae]
peal-xai-rae

[stablediffusion]
captum

[web]
onnx>=1.15
onnxruntime>=1.17
onnx2torch>=1.5
fastapi>=0.110
uvicorn[standard]>=0.27
python-multipart>=0.0.9
omegaconf>=2.3
httpx>=0.27
flask>=2.2
flask-cors>=3.0

[xai]
corelay>=0.2.1
virelay>=0.4.0
zennit-crp>=0.6.0
h5py>=3.7
