[nnx] use jax-style transforms API in nnx_basics #13891
Job | Run time |
---|---|
10s | |
2s | |
12s | |
23s | |
10s | |
2m 59s | |
4m 36s | |
1m 6s | |
2m 23s | |
3m 41s | |
10m 4s | |
4m 5s | |
29m 51s |
Job | Run time |
---|---|
10s | |
2s | |
12s | |
23s | |
10s | |
2m 59s | |
4m 36s | |
1m 6s | |
2m 23s | |
3m 41s | |
10m 4s | |
4m 5s | |
29m 51s |