xla

e6c18f34 - add create_optimized_replica_groups