torch
torchvision
pytest
numpy
torchsummary
gradio
transformers
tiktoken