matplotlib
mlflow~=1.30.0
pytorch-lightning~=1.5.10
omegaconf
seqeval~=1.2.2
scikit-learn
tensorboardx
transformers~=4.26.0
numpy~=1.23.4
datasets~=2.9.0
evaluate~=0.4.0
torchmetrics<0.7  # see https://github.com/PyTorchLightning/pytorch-lightning/issues/11524