talib
torch_optimizer
einops
matplotlib
