First draft
Merge branch 'dump' into balanced_softmax