Distributed Gradient Descent Python