tiktoken
datasets
transformers
wandb
triton==3.2.0
