CoCoNet: Co-Optimizing Computation and Communication for Distributed Machine Learning