Energy balancing in wireless networks with MIMO communications