15Design attention for a 1M-token context on one 8-GPU node: the kernel, the parallelism across GPUs, and the memory plan.▼expertNewGoogle DeepMindAnthropicMeta◆ premiumAt 1M tokens a 70B GQA model needs 320 GiB of KV per request and spends over 90% of prefill FLOPs in attention. The answer lays out a plan that fits on eight H100s, then shows which single change to the model breaks it and what replaces it.Open full answer →
11A 128k-token prefill takes 10 seconds on one node. How does context parallelism split it across nodes, and what does it send?▼hardNewGoogle DeepMindMetaNVIDIA◆ premiumTensor parallelism across nodes would all-reduce every activation over the network. Splitting the sequence instead sends only the KV, which for a GQA model is small. The answer sizes that traffic, finds when it hides behind attention, and fixes the causal-mask imbalance that naive splitting creates.Open full answer →