torch 
transformers 
datasets 
tiktoken 
wandb 
tqdm 
