numpy
einops
tqdm
matplotlib
seaborn

torch
causal-conv1d>=1.4.0
mamba-ssm
tensorboard