Anyhow, I'll try it out. Or do you recommend just using `einsum` itself in Jax, Torch and numpy?
i would suggest just using einsum from the respective module unless you plan on making your pipelines uniform by always using einopt