xla
Allowing `float16` and `bfloat16` computation type and sharding group…
#3617
Merged
Merged