What it does
AI code constantly rearranges grids of numbers called tensors. einops lets programmers describe these changes in short, readable patterns.
It works with PyTorch, JAX, TensorFlow, NumPy, MLX and others, and comes with tutorials.
Who it's for
Programmers and researchers who write AI code.
Can I use it without coding?
Developers only. It is a Python library used in code. New to this? Start here.
Can I run it?
- Cost
- FreeNothing to pay or sign up for.
- Computer
- An ordinary computer is fine
- Privacy
- Runs on your own device: your data stays with you
- API key
- Not needed
Try it
Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?
Before you install
- Only download from the official GitHub page. Copies with similar names exist for many popular projects.
- License (MIT): Very permissive: free to use, change and share, including for business.
- More on staying safe