opt_einsum_fx (conda): Einsum optimization using opt_einsum and PyTorch FX graph rewriting
Keywords: einsum|optimization|pytorch|pytorch-fx