Avoid use of deprecated xla_bridge.get_backend().platform

Co-authored-by: Jake Vanderplas <[email protected]>

Author: rasmi

src/transformers/commands/env.py

@@ -116,7 +116,7 @@ def run(self):
             flax_version = flax.__version__
             jax_version = jax.__version__
             jaxlib_version = jaxlib.__version__
-            jax_backend = jax.lib.xla_bridge.get_backend().platform
+            jax_backend = jax.default_backend()
 
         info = {
             "`transformers` version": version,