Delete non-public API jax.lib.xla_bridge._backends

Jake VanderPlas · Google-ML-Automation · commit e6d6c4ef8a9b · 2024-12-10T13:25:14.000-08:00
This is doubly non-public: nothing under `jax.lib` is public, and also the object itself has a preceding underscore. Therefore it is safe to remove (chex had referenced this previously, but that's now addressed in google-deepmind/chex@adaf1b2). PiperOrigin-RevId: 704825268
diff --git a/jax/lib/xla_bridge.py b/jax/lib/xla_bridge.py
@@ -17,7 +17,6 @@
   default_backend as _deprecated_default_backend,
   get_backend as _deprecated_get_backend,
   xla_client as _deprecated_xla_client,
-  _backends as _backends,
 )
 
 from jax._src.compiler import (

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,6 @@`
`17`	`17`	`default_backend as _deprecated_default_backend,`
`18`	`18`	`get_backend as _deprecated_get_backend,`
`19`	`19`	`xla_client as _deprecated_xla_client,`
`20`		`- _backends as _backends,`
`21`	`20`	`)`
`22`	`21`
`23`	`22`	`from jax._src.compiler import (`