NASA's new AI model can predict when a solar storm may strike