Add support for storing arbitrary PyTrees with Module.perturb()
#518
Job | Run time |
---|---|
7s | |
19s | |
8s | |
18s | |
12s | |
4m 30s | |
3m 49s | |
1m 23s | |
4m 20s | |
3m 57s | |
8m 20s | |
4m 10s | |
31m 33s |
Module.perturb()
#518
Job | Run time |
---|---|
7s | |
19s | |
8s | |
18s | |
12s | |
4m 30s | |
3m 49s | |
1m 23s | |
4m 20s | |
3m 57s | |
8m 20s | |
4m 10s | |
31m 33s |