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

PR from @pzread: fix tinygrad concat + changed formatting to fix CI errors #329

Closed arogozhnikov closed 2 months ago