I implemented this algorithm as part of an assignment for a research course (with extension of a loss function). What the paper doesn't tell you is that it is e

I implemented this algorithm as part of an assignment for a research course (with extension of a loss function). What the paper doesn't tell you is that it is extremely slow, and the memory pressure is insane :/ Really neat paper to go through, though! And not very complicated