Straggler-aware distributed learning: Communication-computation latency trade-off