--index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.org/simple

numpy
pandas
scipy
torch
scikit-learn
tqdm
datasets
gymnasium
transformers[torch]
huggingface-hub
