Coding for straggler mitigation in federated learning