add an einsum_distloss implementation#2

Spark001 wants to merge 3 commits into

sunset1995:mainsunset1995/torch_efficient_distloss:mainfrom

Spark001:einsum_distlossSpark001/torch_efficient_distloss:einsum_distlossCopy head branch name to clipboard