Security for distributed machine learning based software