regex
torch==2.2.1
numpy
wandb