OpenAI has agreed to pay over $20 billion to chip startup Cerebras over the next three years for purchasing server computing power driven by its chips, and may also gain equity in the company as a result.

This is not the first collaboration between the two companies. In January this year, OpenAI signed an agreement to purchase up to 750 megawatts of computing capacity from Cerebras over the next three years, with the transaction valued at more than $10 billion. The new agreement is larger—OpenAI will also provide approximately $1 billion in funding to Cerebras to develop data centers, with total spending potentially reaching $30 billion over the next three years. In return, OpenAI will receive a warrant to purchase a minority stake in Cerebras, and the holding percentage may increase with spending, potentially reaching up to 10% of the shares. Cerebras is expected to disclose some details of the agreement as early as Friday.

For Cerebras, this deep integration with OpenAI is a core support for its push towards going public. The company aims to go public in the second quarter of this year and plans to complete a $3 billion financing round next month, with an estimated valuation of $35 billion at that time.

Cerebras was founded in 2015 and is known for its wafer-scale engine chip technology. Sam Altman, CEO of OpenAI, is one of its early investors. The scale of this deal also reflects the rapid growth in demand for inference computing power in the industry.