Learning Neural Networks with Distribution Shift: Efficiently Certifiable Guarantees