4T transistors, one giant chip (Cerebras WSE-3) [video]

by asdfasdf1on 3/13/2024, 4:13 PMwith 86 comments

by cs702on 3/13/2024, 6:10 PM

According to the company, the new chip will enable training of AI models with up to 24 trillion parameters. Let me repeat that, in case you're as excited as I am: 24. Trillion. Parameters. For comparison, the largest AI models currently in use have around 0.5 trillion parameters, around 48x times smaller.

Each parameter is a connection between artificial neurons. For example, inside an AI model, a linear layer that transforms an input vector with 1024 elements to an output vector with 2048 elements has 1024×2048 = ~2M parameters in a weight matrix. Each parameter specifies by how much each element in the input vector contributes to or subtracts from each element in the output vector. Each output vector element is a weighted sum (AKA a linear combination), of each input vector element.

A human brain has an estimated 100-500 trillion synapses connecting biological neurons. Each synapse is quite a complicated biological structure[a], but if we oversimplify things and assume that every synapse can be modeled as a single parameter in a weight matrix, then the largest AI models in use today have approximately 100T to 500T ÷ 0.5T = 200x to 1000x fewer connections between neurons that the human brain. If the company's claims prove true, this new chip will enable training of AI models that have only 4x to 20x fewer connections that the human brain.

We sure live in interesting times!

---

[a] https://en.wikipedia.org/wiki/Synapse

by brucethemoose2on 3/13/2024, 4:49 PM

Reposting the CS-2 teardown in case anyone missed it. The thermal and electrical engineering is absolutely nuts:

https://vimeo.com/853557623

https://web.archive.org/web/20230812020202/https://www.youtu...

(Vimeo/Archive because the original video was taken down from YouTube)

by fxjon 3/13/2024, 5:25 PM

It has its own programming language CSL

https://www.cerebras.net/blog/whats-new-in-r0.6-of-the-cereb...

"CSL allows for compile time execution of code blocks that take compile-time constant objects as input, a powerful feature it inherits from Zig, on which CSL is based. CSL will be largely familiar to anyone who is comfortable with C/C++, but there are some new capabilities on top of the C-derived basics."

https://github.com/Cerebras/csl-examples

by RetroTechieon 3/13/2024, 6:31 PM

If you were to add up all transistors fabricated worldwide, up until <year>, such that total roughly matches the # on this beast, what year would you arrive? Hell, throw in discrete transistors if you want.

How many early supercomputers / workstations etc would that include? How much progress did humanity make using all those early machines (or any transistorized device!) combined?

by ortusduxon 3/13/2024, 4:38 PM

Not trying to sound critical, but is there a reason to use 4B,000 vs 4T?

by imbusy111on 3/13/2024, 4:50 PM

I wish they dug into how this monstrosity is powered. Assuming 1V and 24kW, that's 24kAmps.

by Rexxaron 3/13/2024, 6:10 PM

Is there a reason it's not roughly a disc if they use the whole wafer ? They could have 50% more surface.

by modelesson 3/13/2024, 10:05 PM

As I understand it, WSE-2 was kind of handicapped because its performance could only really be harnessed if the neural net fit in the on-chip SRAM. Bandwidth to off-chip memory (normalized to FLOPS) was not as high as Nvidia. Is that improved with WSE-3? Seems like the SRAM is only 10% bigger, so that's not helping.

In the days before LLMs 44 GB of SRAM sounded like a lot, but these days it's practically nothing. It's possible that novel architectures could be built for Cerebras that leverage the unique capabilities, but the inaccessibility of the hardware is a problem. So few people will ever get to play with one that it's unlikely new architectures will be developed for it.

by imtringuedon 3/13/2024, 8:35 PM

One thing I don't understand about their architecture is that they have spent so much effort building this monster of a chip, but if you are going to do something crazy, why not work on processing in memory instead? At least for transformers you will primarily be bottlenecked on matrix multiplication and almost nothing else, so you only need to add a simple matrix vector unit behind your address decoder and then almost every AI accelerator will become obsolete over night. I wouldn't suggest this to a random startup though.

by marmadukeon 3/13/2024, 9:57 PM

Hm, let's wait and see what the gemm/W perf is, and how many programmer hours it takes to implement say an mlp. Wafer scale data flow may not be a solved problem?

by tiverton 3/13/2024, 4:56 PM

Interesting. I know there's a lot of attempts to hobble China by limiting their access to cutting edge chips and semiconductor manufacturing technology, but could something like this be a workaround for them, at least for datacenter-type jobs?

Maybe it wouldn't be as powerful as one of these, due to their less capable fabs, but something that's good enough to get the job done in spite of the embargoes.

by asdfasdf1on 3/13/2024, 5:24 PM

WHITE PAPER Training Giant Neural Networks Using Weight Streaming on Cerebras Wafer-Scale Clusters

https://f.hubspotusercontent30.net/hubfs/8968533/Virtual%20B...

by asdfasdf1on 3/13/2024, 5:00 PM

- Interconnect between WSE-2's chips in the cluster was 150GB/s, much lower than NVIDIA's 900GB/s.

- non-sparse fp16 in WSE-2 was 7.5 tflops (about 8 H100s, 10x worse performance per dollar)

Does anyone know the WSE-3 numbers? Datasheet seems lacking loads of details

Also, 2.5 million USD for 1 x WSE-3, why just 44GB tho???

by holodukeon 3/13/2024, 6:37 PM

Better sell all nvidia stocks. Once these chips are common there is no need anymore for GPUs in training super large AI models.

by TradingPlaceson 3/13/2024, 10:05 PM

Near-100% yield is some dark magic.

by apion 3/13/2024, 9:49 PM

I'm surprised we haven't seen wafer scale many-core CPUs for cloud data centers yet.

by beautifulfreakon 3/13/2024, 9:30 PM

So it's increased from 2.6 to 4 trillion transistors over the previous version.

by tedivmon 3/13/2024, 9:35 PM

The missing numbers that I really want to see-

* Power Usage

* Rack Size (last one I played with was 17u)

* Cooling requirements

by tibbydudezaon 3/13/2024, 10:04 PM

Wow - it as bigger than my kitchen tiles - who uses them ???. NSA ???.

by pgrafon 3/13/2024, 9:29 PM

related discussion (2021): https://news.ycombinator.com/item?id=27459466

by hashtag-tilon 3/13/2024, 9:28 PM

Any idea on what’s the yield on these chips?

by wizardforhireon 3/13/2024, 5:56 PM

But can it run doom?

by AdamH12113on 3/13/2024, 4:40 PM

Title should be either "4,000,000,000,000 Transistors" (as in the actual video title) or "4 Trillion Transistors" or maybe "4T Transistors". "4B,000" ("four billion thousand"?) looks like 48,000 (forty-eight thousand).