Adds a minimal but viable implementation of string arrays (with numpy.dtypes.StringDType
) in JAX. Currently this only supports making of a string array by means of either jax.numpy.asarray
or jax.device_put
and reading it back with jax.device_get
.
#21503
+173
−5
Google CLA / cla/google
succeeded
Feb 5, 2025 in 3s
✅ 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:
✅ f278bb2 PR Opener: @copybara-service[bot]
✅ f278bb2 Author: @Google-ML-Automation <googl**********ation@google.com>
(Only the first commit for a unique contributor is listed.)
Loading