The move comes amid rising demand for AI computing power as Google markets its TPUs as rivals to Nvidia hardware.