# prior to this install torch via install.sh

# transformers+
transformers==4.53
accelerate
bitsandbytes
datasets
huggingface-hub[cli]
openai
sentencepiece
tokenizers
trl==0.19  # NOTE as we significantly rely on it we freeze the version but might not be necessary

# pj specific
inspect_ai==0.3.116

# basic
bitarray
ipykernel
ipython
matplotlib
numpy<2.0
pandas
seaborn
scipy
sympy
tqdm
protobuf
pyyaml
tabulate