einops

Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, tensorflow, jax, and others.

More Information

For more information visit the einops website.

Available Versions

These versions of einops are available on Baskerville. These will be retained in accordance with our Applications Support and Retention Policy.