# ------------------------
# PyTorch (CUDA 12.1 builds)
# ------------------------
torch==2.1.0+cu121 --index-url https://download.pytorch.org/whl/cu121
torchvision==0.16.0+cu121 --index-url https://download.pytorch.org/whl/cu121
torchaudio==2.1.0+cu121 --index-url https://download.pytorch.org/whl/cu121

# ------------------------
# PyTorch Geometric wheels
# ------------------------
pyg_lib==0.3.1+pt21cu121 --find-links https://data.pyg.org/whl/torch-2.1.0+cu121.html
torch_cluster==1.6.3+pt21cu121 --find-links https://data.pyg.org/whl/torch-2.1.0+cu121.html
torch_scatter==2.1.2+pt21cu121 --find-links https://data.pyg.org/whl/torch-2.1.0+cu121.html
torch_sparse==0.6.18+pt21cu121 --find-links https://data.pyg.org/whl/torch-2.1.0+cu121.html
torch_spline_conv==1.2.2+pt21cu121 --find-links https://data.pyg.org/whl/torch-2.1.0+cu121.html

# ------------------------
# ML / Data Science utils
# ------------------------
seaborn
ml-collections==0.1.1
tinydb
cvxpy
evotorch
sacred
ogb
wandb
gmpy2
statsmodels
sympy