fix: MultiHeadDotProductAttention and optax ctc_loss changes #24
Job | Run time |
---|---|
4m 7s | |
4m 22s | |
12m 27s | |
3m 8s | |
3m 1s | |
2m 58s | |
5m 30s | |
3m 30s | |
12m 1s | |
3m 13s | |
6m 11s | |
4m 48s | |
1h 5m 16s |
Job | Run time |
---|---|
4m 7s | |
4m 22s | |
12m 27s | |
3m 8s | |
3m 1s | |
2m 58s | |
5m 30s | |
3m 30s | |
12m 1s | |
3m 13s | |
6m 11s | |
4m 48s | |
1h 5m 16s |