pure Java implementaion of Petuum