torch
accelerate
wandb
rich
numpy
autofaiss
transformers
labml_nn