labml>=0.4.60
labml_helpers>=0.4.60
torch
einops
numpy
