FL Games: A federated learning framework for distribution shifts