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
.
#2009
Loading