Avoid use of deprecated device_buffer attriutes of jax.Array
#224
| Job | Run time |
|---|---|
| 33s | |
| 33s |
device_buffer attriutes of jax.Array
#224
| Job | Run time |
|---|---|
| 33s | |
| 33s |