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

Merged

ArthurZucker merged 1 commit intohuggingface:mainfrom hvaara:jax-ndarray

Oct 10, 2023

+28-28

Commits on Oct 9, 2023

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

Roy Hvaara
and
hawkinsp
committed