I see, thanks. For the record, the version I implemented is Eq. (12) (without the regularization term) in http://www.mblondel.org/publications/mblondel-mlj2013.pdf.