einops
на сайте с December 16, 2022 14:10
Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, tensorflow, jax, and others.