VC Vinod Khosla says the US government could take 10% stake in all public companies to soften the blow of AGI