labml==0.4.168
labml-helpers==0.4.89
torch
torchtext
torchvision
einops
numpy
fairscale
