Universal Tensor Operations in Einstein-Inspired Notation for Python
copied from cf-post-staging / einxeinx is a Python library that provides a universal interface to formulate tensor operations in frameworks such as Numpy, PyTorch, Jax and Tensorflow.