changing jnp dtype

[backpropper]

what is the recommended way to change the dtype in jnp array? I'm using something like this:
new_jnp_arr = jnp.asarray(old_jnp_arr, dtype=jnp.int32)

[jakevdp]

Using jnp.asarray with a dtype argument is fine; you can also use new_jnp_arr = old_jnp_arr.astype(jnp.int32). Both approaches end up being implemented in terms of the same lower-level jax.lax.convert_element_type function.

[hawkinsp]

(And note: these mirror NumPy's APIs.)

[backpropper]

I see. I didn't find astype in the jax docs so though it isn't compatible. Thanks!