Closed yfukai closed 2 years ago
Set the JAX version to >=0.3.10 to import jaxlib.xla_extension.XlaRuntimeError. Note XlaRuntimeError will be exposed to jax when https://github.com/google/jax/pull/10676 gets merged.
>=0.3.10
jaxlib.xla_extension.XlaRuntimeError
XlaRuntimeError
jax
Set the JAX version to
>=0.3.10
to importjaxlib.xla_extension.XlaRuntimeError
. NoteXlaRuntimeError
will be exposed tojax
when https://github.com/google/jax/pull/10676 gets merged.