Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
docs: Update JAX backend array_type to jaxlib.xla_extension.ArrayImpl (…
…#2134) * Update JAX backend array_type to jaxlib.xla_extension.ArrayImpl, which is a unified array type introduced in jaxlib v0.4.6. - c.f. https://github.com/google/jax/releases/tag/jaxlib-v0.4.6 - c.f. jax-ml/jax#14768
- Loading branch information