arogozhnikov/einops

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

语言：Python · GitHub 星标：9,630 · Fork：406

话题：cupy · deep-learning · einops · jax · mlx · numpy · pytorch · tensor

在 GitHub 上查看原仓库