For example, NASH (Shen et al., 2018) passes the gradients through the Bernoulli stochastic layer using ST estimator.
they do: where the ST estimator lets the gradients of the stochastic layer dz dg φ (x) ≈ 1.