python3-einx

Universal Tensor Operations in Einstein-Inspired Notation

Description

einx is a Python library that provides a universal interface to formulate tensor operations in frameworks such as Numpy, PyTorch, Jax and Tensorflow. The design is based on the following principles:

Provide a set of elementary tensor operations following Numpy-like naming:
 einx.{sum|max|where|add|dot|flip|get_at|...}

Use einx notation to express vectorization of the elementary operations. einx notation is inspired by einops, but introduces several novel concepts such as []-bracket notation and full composability that allow using it as a universal language for tensor operations. einx can be integrated and mixed with existing code seamlessly. All operations are just-in-time compiled into regular Python functions using Python's exec() and invoke operations from the respective framework.

Upload more screenshots

Please help extend the collection of screenshots. Just make a screenshot and upload it here. You don't need to register or anything.

Upload a screenshot

Hint: upload an image here from your clipboard with Ctrl-V


Homepage

https://github.com/fferflo/einx


Install this software package

If the package is available for the distribution you are currently using on your computer then install the software by clicking on…

Install python3-einx