discrete-key-value-bottleneck-pytorch
einops>=0.3
numpy
torch>=1.6
torchmetrics
polars
pyfaidx
pyyaml
transformers
