torch
pytorch-lightning
wandb
transformers
datasets
tokenizers
sacrebleu
sacremoses
