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
.
#23678
Job | Run time |
---|---|
2m 17s | |
2m 15s | |
2m 17s | |
6m 49s |