Click>=7.0
tqdm
pyyaml>=5.1
torch
torch_optimizer
pytorch-lightning
sparsemax
numpy
