xla

057e0b70 - Remove hacky copy_ registration & added a backward test for einsum. (#2401)