lightning
transformers
datasets
wandb
rich
tensorboard
