Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Why do you think that https://github.com/dfm/extending-jax is a good alternative to einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Why do you think that https://github.com/dfm/extending-jax is a good alternative to einops