[x64 deprecation] Create _np_array utility routine (#3779)
[x64 deprecation] Create _np_array utility routine (#3779)


tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax,google/jax