Most infants born at 36 and 37 weeks' gestation are mature enough to be discharged from the hospital with the mother. But many premature infants need care in the neonatal intensive care unit (NICU). Hospital care will be needed for: 1  Underdeveloped lungs. 2  Inability to breathe continuously (apnea of prematurity) . 3  Inability to maintain body heat.