torch
transformers
torchdata
datasets