carabiner-tools[mpl,pd]>=0.0.4
datasets<4,>=3.0.0
duvida[torch]>=0.0.3.post1
huggingface_hub
numpy
platformdirs>=4.4.0
torch<=2.9,>=2.4
torchvision
lightning
scipy
tensorboard
tqdm

[bio]
vectome>=0.0.4

[chem]
chemprop>=2.1.0
schemist>=0.0.4.post1
scikit-learn>=1.7

[dev]
duvidnn[bio,chem,splits,transformers]
pytest
flake8

[splits]
umap-learn
scipy
schemist>=0.0.4.post1
tdigest>=0.5
faiss-cpu>=1.11

[splits_gpu]
umap-learn
scipy
schemist>=0.0.4.post1
tdigest>=0.5
faiss-gpu

[transformers]
torch>=2.4
torchvision
transformers
