xla

Remove hacky copy_ registration & added a backward test for einsum.

#2401

Merged