The interaction between `include_prob` and `mask_weights` should be described: a weight will be non-zero only if it is included (sampled with include_prob) _and_ its mask_weights is non-zero.