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

[dev]
jupyterlab
