[Date Prev][Date Next] [Thread Prev][Thread Next] [Date Index] [Thread Index]

Bug#1111853: ITP: python-einx -- Universal Tensor Operations in Einstein-Inspired Notation



Package: wnpp
Severity: wishlist
Owner: Steffen Moeller <moeller@debian.org>
X-Debbugs-Cc: debian-devel@lists.debian.org, moeller@debian.org

* Package name    : python-einx
* URL             : https://github.com/fferflo/einx
* License         : MIT
  Programming Lang: Python
  Description     : Universal Tensor Operations in Einstein-Inspired Notation

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.

The package hopes to live on
https://salsa.debian.org/python-team/packages/python-einx .


Reply to: