torch
transformers
pickle