[compatible with jax 0.4.36] remove ensure_compile_time_eval context, fix progress bar in loop transformation
#249
| Job | Run time |
|---|---|
| 2m 24s | |
| 2m 16s | |
| 3m 1s | |
| 2m 14s | |
| 3m 5s | |
| 3m 11s | |
| 1m 55s | |
| 2m 17s | |
| 3m 14s | |
| 2m 22s | |
| 2m 28s | |
| 28m 27s |