jax

5be30dd5 - Make CPU devices report as cpu:0 instead of TFRT_CPU_0.

Make CPU devices report as cpu:0 instead of TFRT_CPU_0. The "TFRT" is vestigial and confusing. The new string matches how we report GPUs (e.g., cuda:0). PiperOrigin-RevId: 890641012