Skip to content

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. #1838

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.

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. #1838

Triggered via pull request January 21, 2025 18:38
Status Success
Total duration 7m 32s
Artifacts

cloud-tpu-ci-presubmit.yml

on: pull_request
Matrix: cloud-tpu-test
Fit to window
Zoom out
Zoom in