Cerebras WSE-3: Third Era Superchip for AI

Sunnyvale, Calif., AI supercomputer agency Cerebras says its subsequent technology of waferscale AI chips can do double the efficiency of the earlier technology whereas consuming the identical quantity of energy. The Wafer Scale Engine 3 (WSE-3) incorporates 4 trillion transistors, a greater than 50 p.c improve over the earlier technology due to using newer chipmaking know-how. The corporate says it should use the WSE-3 in a brand new technology of AI computer systems, which are actually being put in in a datacenter in Dallas to kind a supercomputer able to 8 exaflops (8 billion billion floating level operations per second). Individually, Cerebras has entered right into a joint improvement settlement with Qualcomm that goals to spice up a metric of worth and efficiency for AI inference 10-fold.

The corporate says the CS-3 can practice neural community fashions as much as 24-trillion parameters in dimension, greater than 10 instances the scale of as we speak’s largest LLMs.

With WSE-3, Cerebras can hold its declare to producing the most important single chip on this planet. Sq.-shaped with 21.5 centimeters to a facet, it makes use of almost a whole 300-millimeter wafer of silicon to make one chip. Chipmaking tools is often restricted to producing silicon dies of not more than about 800 sq. millimeters. Chipmakers have begun to flee that restrict by utilizing 3D integration and different superior packaging know-how3D integration and different superior packaging know-how to mix a number of dies. However even in these methods, the transistor depend is within the tens of billions.

As regular, such a big chip comes with some mind-blowing superlatives.

You’ll be able to see the impact of Moore’s Legislation within the succession of WSE chips. The primary, debuting in 2019, was made utilizing TSMC’s 16-nanometer tech. For WSE-2, which arrived in 2021, Cerebras moved on to TSMC’s 7-nm course of. WSE-3 is constructed with the foundry large’s 5-nm tech.

The variety of transistors has greater than tripled since that first megachip. In the meantime, what they’re getting used for has additionally modified. For instance, the variety of AI cores on the chip has considerably leveled off, as has the quantity of reminiscence and the interior bandwidth. Nonetheless, the advance in efficiency by way of floating-point operations per second (flops) has outpaced all different measures.

CS-3 and the Condor Galaxy 3

The pc constructed across the new AI chip, the CS-3, is designed to coach new generations of large giant language fashions, 10 instances bigger than OpenAI’s GPT-4 and Google’s Gemini. The corporate says the CS-3 can practice neural community fashions as much as 24-trillion parameters in dimension, greater than 10 instances the scale of as we speak’s largest LLMs, with out resorting to a set of software program tips wanted by different computer systems. In line with Cerebras, meaning the software program wanted to coach a one-trillion parameter mannequin on the CS-3 is as easy as coaching a one billion parameter mannequin on GPUs.

As many as 2,048 methods might be mixed, a configuration that may chew by way of coaching the favored LLM Llama 70B from scratch in simply in the future. Nothing fairly that massive is within the works, although, the corporate says. The primary CS-3-based supercomputer, Condor Galaxy 3 in Dallas, shall be made up of 64 CS-3s. As with its CS-2-based sibling methods, Abu Dhabi’s G42 owns the system. Along with Condor Galaxy 1 and a couple of, that makes a community of 16 exaflops.

“The present Condor Galaxy community has skilled among the main open-source fashions within the business, with tens of 1000’s of downloads,” mentioned Kiril Evtimov, group CTO of G42 in a press launch. “By doubling the capability to 16 exaflops, we sit up for seeing the subsequent wave of innovation Condor Galaxy supercomputers can allow.”

A Deal With Qualcomm

Whereas Cerebras computer systems are constructed for coaching, Cerebras CEO Andrew Feldman says it’s inference, the execution of neural community fashions, that’s the actual restrict to AI’s adoption. In line with Cerebras estimates, if each particular person on the planet used ChatGPT, it could price US $1 trillion yearly—to not point out an amazing quantity of fossil-fueled vitality. (Working prices are proportional to the scale of neural community mannequin and the variety of customers.)

So Cerebras and Qualcomm have shaped a partnership with the aim of bringing the price of inference down by an element of 10. Cerebras says their answer will contain making use of neural networks strategies similar to weight information compression and sparsity—the pruning of unneeded connections. The Cerebras-trained networks would then run effectively on Qualcomm’s new inference chip, the AI 100 Extremely, the corporate says.