xla

1db6c861 - Manually register einsum on xla (#8801)