torch
torchinfo
transformers
huggingface
