Zaslat SMS: Gradient matching for domain generalization