Allow withgradient to pass aux info, and add @grad dropout
#94
| Job | Run time |
|---|---|
| 2m 38s | |
| 2m 42s | |
| 2m 25s | |
| 3m 54s | |
| 11m 39s |
withgradient to pass aux info, and add @grad dropout
#94
| Job | Run time |
|---|---|
| 2m 38s | |
| 2m 42s | |
| 2m 25s | |
| 3m 54s | |
| 11m 39s |