Fix PRNG handling in nn.jit
under nn.scan
.
#558
Job | Run time |
---|---|
20s | |
5s | |
14s | |
15s | |
6s | |
4m 25s | |
4m 10s | |
1m 37s | |
4m 18s | |
3m 59s | |
8m 14s | |
3m 56s | |
31m 39s |
nn.jit
under nn.scan
.
#558
Job | Run time |
---|---|
20s | |
5s | |
14s | |
15s | |
6s | |
4m 25s | |
4m 10s | |
1m 37s | |
4m 18s | |
3m 59s | |
8m 14s | |
3m 56s | |
31m 39s |