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.
#70409
| Job | Run time |
|---|---|
| 8m 50s | |
| 8m 44s | |
| 1m 51s | |
| 3m 0s | |
| 1m 14s | |
| 1m 19s | |
| 2m 23s | |
| 27m 21s |