arogozhnikov / einops

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

Home Page:https://einops.rocks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

arogozhnikov/einops Stargazers