numpy
torchrua>=0.2.0
torch-scatter

[benchmark]
aku
tqdm
pytorch-crf

[dev]
einops
pytest
hypothesis
pytorch-crf
