Feinsum, short for Fast-Einstein Summations, provides tools for managing databases of code-transformations on Batched Einstein Summations. Unlike typical computational primitive libraries that ship optimized computational kernel in source or binary form, feinsum records code-transformations from a trivial schedule, resulting in relative savings from not having to flush the operands to the DRAM.
- Install as
pip install feinsum
. - Read the documentation at https://feinsum.readthedocs.io.
- Source-code, Issue Management at https://github.com/kaushikcfd/feinsum.