xla

[benchmarks] Default to `bfloat16` (inference) and AMP (training) precision.

#6518

Merged