SAE on toy model (with resampling)