The jaw-dropping energy demands of AI tools has led some California companies and lawmakers to re-examine the possibility of nuclear power.