Implement BalancingLearner

@jbweston or @anton-akhmerov

It would be nice if you guys could look at this, I am going nuts with finding the bug.

Sometimes the snippet in the notebook works without issues, but other times there is an error that I am unable to reproduce.

Closes #10 (closed), #13 (closed)