Yes, that’s how it works (pipeline parallelism)

mg · on July 16, 2024

Interesting. Let's do the math ...

Let's say the model has 50B parameters and 50 layers. That would mean about one billion values have to travel through the wifi for every generated token?

I wonder how much data that is in bytes and how long it takes to transfer them.

blackbear_ · on July 16, 2024

It's not the parameters that are sent, it's the layer outputs. That makes for a few thousands floats per token

mg · on July 16, 2024

Woops! I would have thought the number of neurons roughly equals the number of parameters, but you are right. The number of parameters is much higher.

tama_sala · on July 16, 2024

The embedding size is only 8k so while the parameters are 70B. So it's a huge difference