Hi, I quite don't understand where `Eq. (4), (7)` of the paper come in this implementation. Can you point me to it? Thanks