FICHA · AUR

python-einops

Deep learning operations reinvented (for pytorch, tensorflow, jax and others)

  • Tensor operation library
  • LIBRARY
  • AI
  • Dependency only
official+codex · reviewed · Jun 3, 2026 description in en

Description

Tensor reshaping and reduction patterns can be expressed consistently across PyTorch, TensorFlow, JAX, and other array stacks. Machine-learning developers use it to make model code shorter and clearer. Shape mistakes can silently change model behavior, so tests around tensor dimensions still matter.

Permissions

Permissions not analysed for this source yet.