[JAX] Replace uses of jnp.array
in types with jnp.ndarray
.
#1173
Job | Run time |
---|---|
3m 12s | |
3m 29s | |
6m 41s |
jnp.array
in types with jnp.ndarray
.
#1173
Job | Run time |
---|---|
3m 12s | |
3m 29s | |
6m 41s |