How they’re using AI at the lab that created the atom bomb.