Fix PRNG handling in nn.jit
under nn.scan
.
#557
Job | Run time |
---|---|
7s | |
26s | |
4s | |
15s | |
12s | |
0s | |
1m 4s |
nn.jit
under nn.scan
.
#557
Job | Run time |
---|---|
7s | |
26s | |
4s | |
15s | |
12s | |
0s | |
1m 4s |