Added device to jnp.arange
, jnp.eye
and tests
#307
Job | Run time |
---|---|
47s | |
47s |
jnp.arange
, jnp.eye
and tests
#307
Job | Run time |
---|---|
47s | |
47s |