Tensor train representation of MIMO channels using the JIRAFE method