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