opt-einsum-fx (pypi): Einsum optimization using opt_einsum and PyTorch FX
Keywords: einsum|optimization|pytorch|pytorch-fx