Lkjcorr returning error

That by itself is not a limitation if the LKJ also has a batch dimension (groups in your case)