numpy == 1.21.5
scipy == 1.8.0
einops == 0.4.1