The code is already available on nr-master. We are still working on it, so there may be unknown bugs, on top of the known limitations.
As for the release date, we need to wait for ns-3.48 release (end of May) which includes patches required not to break the simulation due to precoding matrix dimension mismatches.
Otherwise you can only use SISO, which defeats the purpose.