--extra-index-url https://download.pytorch.org/whl/cpu

torch
torchvision
torchmetrics
pytorch-lightning
accelerate
transformers
scikit-learn
matplotlib
seaborn
bayesian-optimization
stable-baselines3
openai
reformer-pytorch