requirements.txt

aiohappyeyeballs==2.4.3
aiohttp==3.10.10
aiosignal==1.3.1
annotated-types==0.7.0
async-timeout==4.0.3
attrs==24.2.0
Automat==24.8.1
axial-positional-embedding==0.2.1
buildtools==1.0.6
causal-conv1d==1.2.0.post2
certifi==2022.12.7
charset-normalizer==2.1.1
constantly==23.10.4
deepspeed==0.15.3
docopt==0.6.2
einops==0.8.0
filelock==3.16.1
frozenlist==1.5.0
fsspec==2024.10.0
furl==2.1.3
greenlet==3.1.1
hjson==3.1.0
huggingface-hub==0.26.2
hyperlink==21.0.0
idna==3.4
incremental==24.7.2
Jinja2==3.1.4
joblib==1.4.2
lightning-utilities==0.11.8
local-attention==1.9.15
mamba-ssm==1.2.0.post1
MarkupSafe==3.0.2
msgpack==1.1.0
multidict==6.1.0
networkx==3.4.2
ninja==1.11.1.1
numpy==1.26.3
nvidia-ml-py==12.560.30
ogb==1.3.6
openbabel-wheel==3.1.1.20
orderedmultidict==1.0.1
packaging==24.1
pandas==2.2.3
performer-pytorch==1.1.4
pillow==10.2.0
propcache==0.2.0
psutil==6.1.0
py-cpuinfo==9.0.0
pydantic==2.9.2
pydantic_core==2.23.4
pyg-lib==0.4.0+pt113cu117
pyg-nightly==2.7.0.dev20241106
pyparsing==3.2.0
python-dateutil==2.9.0.post0
pytorch-lightning==2.1.0
pytz==2024.2
PyYAML==6.0.2
rdkit==2024.3.5
redo==3.0.0
regex==2024.9.11
requests==2.28.1
safetensors==0.4.5
scikit-learn==1.5.2
scipy==1.14.1
simplejson==3.19.3
six==1.16.0
SQLAlchemy==2.0.36
tensorboardX==2.6.2.2
threadpoolctl==3.5.0
tokenizers==0.20.2
tomli==2.0.2
torch==1.13.0+cu117
torch-cluster==1.6.3
torch-geometric==2.6.0
torch-scatter==2.1.2
torch-sparse==0.6.18
torch-spline-conv==1.2.2+pt113cu117
torchaudio==0.13.0+cu117
torchmetrics==1.5.1
torchvision==0.14.0+cu117
tqdm==4.66.6
transformers==4.46.1
triton==3.1.0
Twisted==24.10.0
typing_extensions==4.9.0
tzdata==2024.2
urllib3==1.26.13
wandb==0.18.5
yacs==0.1.8
yarl==1.17.1
zope.interface==7.1.1