A Java implementation of Petuum