torch
colossalai
titans
pulp
datasets
matplotlib
transformers