Tensor parallelism
Split every weight matrix across GPUs: column and row splits with real numbers, a TP transformer block, and what it costs in communication.
Projects and perspectives from AI Next Labs
Start with one document. There’s more than one idea inside.
This is a window into the work at AI Next Labs. We share the things we build, the ideas we’re exploring, and what we learn along the way.
Read our perspectivesExplore concepts our team works with, explained through interactive guides.
Our perspectives on AI, the topics we’re discussing, and lessons from our work.
Projects from the team, with a closer look at how they were made.
Take an idea further. Guides and resources from the things we share.
Turn a document into searchable chunks. Find the right facts. Give them to the AI to help it answer.
Split every weight matrix across GPUs: column and row splits with real numbers, a TP transformer block, and what it costs in communication.
In 2016, the godfather of AI told the world to stop training radiologists. A decade later the opposite had happened, and the shape of his mistake is the most useful map we have for what to become in the age of AI.
Keep an editable copy, save a PDF from Google Docs, and check page breaks, tables and equations before sharing.
Copy the model, split the batch: ring all-reduce, overlapping communication, the batch-size recipe and ZeRO stages 1 to 3.
Transformer shapes, the 16 bytes per parameter, activation memory, FLOPs, gradient accumulation and recomputation, with step-through diagrams.
In about 5 minutes, review remembered details, switch memory off, remove one saved memory, and start a Temporary Chat.