Backend JAX: learning rate decay with Optax (#1992) #1541
| Job | Run time |
|---|---|
| 2m 52s | |
| 2m 51s | |
| 1m 49s | |
| 1m 44s | |
| 6m 34s | |
| 4m 53s | |
| 2m 59s | |
| 6m 24s | |
| 1m 38s | |
| 2m 57s | |
| 1m 30s | |
| 7m 5s | |
| 43m 16s |
| Job | Run time |
|---|---|
| 2m 52s | |
| 2m 51s | |
| 1m 49s | |
| 1m 44s | |
| 6m 34s | |
| 4m 53s | |
| 2m 59s | |
| 6m 24s | |
| 1m 38s | |
| 2m 57s | |
| 1m 30s | |
| 7m 5s | |
| 43m 16s |