A Federated Cox Model with Non-Proportional Hazards