einops local-attention matplotlib numpy pandas patool reformer-pytorch scikit-learn scipy sktime sympy torch tqdm PyWavelets