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