einops
numpy
matplotlib