Further changes to use jax.extend.core
instead of jax.core
.
#2324
Job | Run time |
---|---|
6m 29s | |
6m 25s | |
12m 54s |
jax.extend.core
instead of jax.core
.
#2324
Job | Run time |
---|---|
6m 29s | |
6m 25s | |
12m 54s |