Allows `KerasMetricCallback` to use XLA generation
Updates the `KerasMetricCallback` with the ability to use XLA generation for a big speed boost! cc @merveenoyan 
_The documentation is not available anymore as the PR was closed or merged._