torch==1.12.1+cu116
transformers==4.29.1
einops
tqdm
wandb
rotary-embedding-torch
biopython
scipy
dm-tree
matplotlib
seaborn
black
ipython
--extra-index-url https://download.pytorch.org/whl/cu116