Workaround to make XRT work with multi-backend change.
Workaround to make XRT work with multi-backend change.

XrtBuffers don't expose their platform name. I couldn't figure out a nice way to plumb this through, so I added this workaround for now.
google/jax,tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax