Backend JAX: learning rate decay with Optax #1537
| Job | Run time |
|---|---|
| 5m 18s | |
| 1m 29s | |
| 2m 18s | |
| 2m 8s | |
| 2m 50s | |
| 2m 51s | |
| 5m 41s | |
| 2m 56s | |
| 1m 51s | |
| 2m 56s | |
| 5m 36s | |
| 5m 45s | |
| 41m 39s |
| Job | Run time |
|---|---|
| 5m 18s | |
| 1m 29s | |
| 2m 18s | |
| 2m 8s | |
| 2m 50s | |
| 2m 51s | |
| 5m 41s | |
| 2m 56s | |
| 1m 51s | |
| 2m 56s | |
| 5m 36s | |
| 5m 45s | |
| 41m 39s |