diffusers

[JAX] Replace uses of `jnp.array` in types with `jnp.ndarray`.

#4719

Merged