Trump administration in talks for equity stakes in AI companies