numpy<2,>=1.25
pandas<2.2
pyarrow<15
scikit-learn
scipy
rdkit
tqdm
transformers==4.36.2
safetensors
tokenizers
accelerate
torch<2.3,>=2.1.2
torchvision<0.17,>=0.16
torchaudio<2.2,>=2.1

[dev]
jupyterlab
