/

cupy

arogozhnikov/einops
503日前7.7k

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)