mention that cupy and jax now support array api

arogozhnikov / einops

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

https://einops.rocks

MIT License

8.55k stars 352 forks source link

Closed arogozhnikov closed 1 week ago