xla

c55bfb80 - Initial support for all_reduce