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

[:sys_platform == "linux"]
flash-attn

[beignet_mdtraj]
beignet[mdtraj]

[mgm]
rdkit>=2024.9.4
selfies>=2.1.2
