Enabling stratified sampling in high dimensions via nonlinear dimensionality reduction