IMC 2019 Problem 9

Determine all positive integers $n$ for which there exist $n\times n$ real invertible matrices $A$ and $B$ that satisfy $AB-BA=B^2A$ .

Proposed by Karen Keryan, Yerevan State University & American University of Armenia, Yerevan