change numpy to jax-numpy to enable grad functionality #9
Job | Run time |
---|---|
7s | |
5m 4s | |
5m 12s | |
4m 57s | |
4m 49s | |
3m 55s | |
5m 38s | |
5m 12s | |
5m 25s | |
5m 22s | |
0s | |
45m 41s |
Job | Run time |
---|---|
7s | |
5m 4s | |
5m 12s | |
4m 57s | |
4m 49s | |
3m 55s | |
5m 38s | |
5m 12s | |
5m 25s | |
5m 22s | |
0s | |
45m 41s |