Raise a runtime error when trying to convert the jax.Array
wrapped by jax.core.Token
to a numpy array, as it is an internal implementation detail and the buffer has XLA token shape.
#5531
Loading