rewrite _make_jaxpr() function to be compatible with jax==0.4.29
#16
| Job | Run time |
|---|---|
| 1m 7s | |
| 1m 23s | |
| 3m 10s | |
| 1m 17s | |
| 1m 15s | |
| 2m 25s | |
| 1m 15s | |
| 56s | |
| 2m 38s | |
| 1m 28s | |
| 1m 26s | |
| 2m 23s | |
| 20m 43s |