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