pooch
torch
tqdm
lightning
torchmetrics
biopython
torchvision
universal_pathlib
pandas
scipy
pyarrow
s3fs[boto3]
biopandas
wandb
hydra-core
torchdata
datasets>2
tokenizers
transformers>=4.24.0
scikit-learn
python-dotenv
torcheval
datasketch
peft
icecream
captum
edlib
onnx
onnxscript
einops
beignet[datasets]
deepspeed
dotenv>=0.9.9
umap-learn>=0.5.7

[flash]

[flash:sys_platform == "linux"]
flash-attn>=2.7.4.post1

[mgm]
rdkit>=2024.9.4
selfies>=2.1.2
