The company saw improved demand for chips in the wake of its partnership with Nvidia and an investment from the U.S. government.