Given the condition $AB = BA$, the expression $A^2 + B^2$ can be simplified. \[A^2 + B^2 = (A + B)^2 - 2AB\] Due to the commutativity of $A$ and $B$ (i.e., $AB = BA$), the simplification proceeds as follows:\[A^2 + B^2 = (A + B)^2 - 2AB = 2(A + B)\] Therefore, the final simplified form is $2(A + B)$.