carabiner-tools[mpl,pd]>=0.0.4
datasets>=2.15.0
huggingface_hub
numpy
tqdm

[chem]
chemprop>=2.1.0
schemist>=0.0.4.post1
lightning
tensorboard

[jax]
jax>=0.5.3

[jax_cuda12]
jax[cuda12]>=0.5.3

[jax_cuda12_local]
jax[cuda12_local]>=0.5.3

[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

[torch]
torch>=2.4
torchvision
lightning
tensorboard

[transformers]
torch>=2.4
torchvision
transformers
