numpy
transformers 
datasets 
tiktoken 
wandb 
tqdm
torch