arogozhnikov

arogozhnikov/einops

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

9.5K stars 400 forks 9.5K watchers Python MIT License
cupy deep-learning einops jax mlx numpy pytorch tensor tensorflow
View on GitHub Website
1 Open Issue Need Help Last updated: Jul 5, 2026

Open Issues Need Help

View All on GitHub
Missing einops.layers.....Repeat about 2 hours ago
good first issue feature suggestion
arogozhnikov/einops
9.5K

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

Python
#cupy#deep-learning#einops#jax#mlx#numpy#pytorch#tensor#tensorflow