labml
torch
torchtext
torchvision
einops
numpy
fairscale
