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.
#67308
Google CLA / cla/google
succeeded
May 11, 2024 in 2s
✅ All contributors are covered under a CLA with Google
See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).
ℹ️ Googlers: Go here to view more details and manage scans for this pull request.
Details
The following contributors were found for this pull request:
✅ 800f114 PR Opener: @copybara-service[bot]
✅ 800f114 Author: @yueshengys <yue*****ys@google.com>
(Only the first commit for a unique contributor is listed.)
Loading