jax

88cf87c1 - Remove deprecated APIs jax.lib.xla_extension.ArrayImpl and XlaRuntimeError.

Remove deprecated APIs jax.lib.xla_extension.ArrayImpl and XlaRuntimeError. These were deprecated in v0.5.0 and removed in v0.6.0. Accessing them currently results in a custom AttributeError; after this change it will result in a generic AttributeError.