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
.
#26141
Job | Run time |
---|---|
2m 2s | |
2m 0s | |
1m 55s | |
5m 57s |