Three Ways to Shard AI Models and Their Tradeoffs

A short guide to weight, input and pipeline parallelism, and the costs of splitting computation.

Channel
Cognitive Revolution "How AI Changes Everything"
Published
Jun 4, 2024
Video
1:00
Read
1 min
Original
English
Open on YouTube
Scroll to read and the video shrinks to the bottom-right corner. Captions follow your reading language.
Reading languageAvailable in 2 languages
You're reading the original English.
SUMMARY

In 30 seconds

A speaker on The Cognitive Revolution compares three approaches to sharding AI models, highlighting their effects on model size, latency and implementation complexity.

  1. 1Three approachesSplit weights, input activations or model stages.
  2. 2Weight sharding has a costTensor parallelism can help size and latency, but introduces overhead.
  3. 3Attention complicates input shardingSplitting input requires care because tokens attend to one another.

Takeaways

  1. Model weights, input activations and pipeline stages can each be split.
  2. Weight sharding can help latency and model size but adds overhead.
  3. Input sharding complicates attention; the speaker recommends minimizing unnecessary sharding.
FULL TEXT

Read the edited text by chapter

Transcript only
01

Ways to shard

Compare weight, activation and pipeline approaches.

There are three broad ways to shard a model computation: split the model weights, split the input activations, or divide the model into stages, known as pipeline parallelism. The speaker says pipeline parallelism helps when a model is too large, but does not itself improve latency.

Sharding the weights, also called model or tensor parallelism, can address model size and help latency. But sharding has a cost, so there is a performance tradeoff.

02

Costs and attention

Input splitting complicates attention; shard as much as needed.

Input sharding is relatively cheap, but usually does not solve the model-size problem. Attention requires extra care: tokens attend to other tokens, so splitting them introduces additional complexity.

The speaker’s practical advice is to shard only as much as necessary. Sharding brings overhead, but some degree of it can be unavoidable when a model cannot otherwise fit.

More from this channel

1:49:40EN
Cognitive Revolution "How AI Changes Everything"· Oct 28, 2026Podcast · 4 speakersA Strategy for Preserving Options on the Path to Advanced AIGuest Joel Predd, Ben Boudreaux, Matt Chessen · Host Nathan LabenzRAND researchers discuss uncertainty, security architecture and human agency with Nathan Labenz.AI StrategyGeopoliticsAI Safety+1
1:28:36EN
Cognitive Revolution "How AI Changes Everything"· Oct 24, 2026Podcast · 5 speakersAI’s Pace: Interpretability, GPU Credit and the Physical Supply ChainGuest Zvi Mowshowitz, Evan Conrad, Lonis Hamaili · Host Nathan Labenz, Prakash NarayananA weekly highlights episode weighs uncertain AI research trajectories against finance, infrastructure, security and regulation.AI researchAI SafetyAI Regulation
3:13:44EN
Cognitive Revolution "How AI Changes Everything"· Oct 22, 2026Podcast · 4 speakersAI’s Physical Bottlenecks: Compute Markets, Chip Investments and Agent TrustGuest Evan Conrad, Lonis Hamaili · Host Nathan Labenz, Prakash NarayananA full AI:AM discussion of security incidents, GPU financing, semiconductor supply and an agent-to-agent network.Infrastructure & deploymentAI securityInvesting+2
2:40:58EN
Cognitive Revolution "How AI Changes Everything"· Oct 21, 2026Podcast · 3 speakersFrontier AI, Interpretability and the Fight Over Its PaceGuest Zvi Mowshowitz · Host Nathan Labenz, Prakash NarayananZvi Mowshowitz joins AI:AM to discuss neural networks, policy and the institutions facing rapid change.AI SafetyAI RegulationAI research+2
1:25:56EN
Cognitive Revolution "How AI Changes Everything"· Oct 17, 2026PodcastClaude Changed the Password. The Steering Wheel Was Already OptionalHost Nathan LabenzNathan Labenz and Prakash Narayanan spend a week on agents that can lock you out, models that dodge a feeling they deny having, and formal proofs that found Linux bugs.AgentsSafety & governanceAgent engineering+3
3 slides2:51:22EN
Cognitive Revolution "How AI Changes Everything"· Oct 14, 2026Podcast · 4 speakersAI Minds and 8 mm Memories: Research, Archives and SimulationGuest Cameron Berg, Phil Maher · Host Nathan Labenz, Prakash NarayananCameron Berg discusses hidden model states; Phil Maher explains preserving home movies and a fly-brain demo.ConsciousnessAI researchAI Safety+2