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
.
#5794
Triggered via pull request
January 27, 2025 23:06
copybara-service[bot]
synchronize
#21503
Status
Skipped
Total duration
3s
Artifacts
–