Pass grad_estimator
to the CSGLD kernel directly
#518
+20
−8
Merged