Packed Sequences#

Packed sequences are a fine-tuning technique that reduces padding waste by concatenating multiple examples into one pack while preserving sequence boundaries for attention. In Megatron Bridge, this is primarily a supervised fine-tuning and PEFT optimization rather than a general pretraining feature.

This page is the stable overview for what packed sequences are, when to use them, and which constraints are durable. For operational setup, code anchors, and verification commands, see skills/nemo-mbridge-perf-sequence-packing/SKILL.md.

What It Is#

Fine-tuning datasets often contain examples with highly variable lengths. When those examples are batched conventionally, many tokens in each batch are just padding. Packed sequences reduce that waste by building longer packs from multiple examples and carrying boundary metadata into the attention path.

In Bridge today, there are two distinct packing paths plus long-context enablement through context parallelism:

Path

Use case

Key config

Offline packed SFT

Text-only finetuning

enable_offline_packing=True plus offline_packing_specs

Direct-HF/VLM in-batch packing

Direct Hugging Face and supported VLM finetuning

enable_in_batch_packing=True

Long-context (CP)

Pretrain / finetune at 16K-128K+

context_parallel_size > 1

These are related but they are not the same knob. Offline packed SFT and Direct-HF/VLM in-batch packing solve padding waste; long-context training primarily addresses activation memory and communication tradeoffs at larger sequence lengths.

The shared implementation lives under megatron.bridge.data.packing: offline GPT SFT materialization, packed Parquet runtime datasets, bin-packing algorithms, and collate-time THD packing each have separate modules. Ordinary non-packed padding remains in megatron.bridge.data.collators. Use scripts/training/prepare_gpt_sft_packed_data.py when packed GPT SFT artifacts should be prepared before launching training.

When to Use It#

Packed sequences are a good fit when all of the following are true:

  • you are doing SFT, PEFT, or supported VLM finetuning using one of the two packing paths above

  • your examples have variable lengths and padding waste is significant

  • you can tolerate the micro-batch constraints of packed training

Packed sequences are usually not the right answer when:

  • you are doing standard Megatron-style pretraining, which already concatenates documents during sampling

  • you want long-context training in general, where context parallelism is often the main technique

  • your model family or recipe explicitly opts out of packed-sequence support

Stable Constraints#

The durable constraints for packed sequences in Bridge are:

  • offline packed SFT requires configured micro_batch_size == 1

  • Direct-HF/VLM in-batch packing requires configured micro_batch_size > 1; collation flattens those input rows into one physical THD batch row

  • when context parallelism is used, sequence length must satisfy the standard CP divisibility constraints

  • Direct-HF sequence length must also satisfy the LCM of the training and evaluation CP constraints and CP * TP when sequence parallelism is enabled

  • for fine-tuning with CP enabled, per-token loss behavior and reduction settings matter

  • CUDA-graph-friendly packed metadata requires additional padding constraints

Model-family support is not universal. Some families and recipe paths explicitly opt out of packed sequences or related packing modes.

Relationship to Long-Sequence Training#

Packed sequences and long-sequence training are often mentioned together because both affect sequence layout and memory behavior, but they solve different problems:

  • packed sequences mainly reduce padding waste in fine-tuning datasets

  • long-sequence training mainly addresses activation memory and communication tradeoffs at larger sequence lengths

For long-sequence training guidance, see:

  • docs/performance-guide.md

  • docs/training/hierarchical-context-parallel.md

Practical Caveats#

The most stable caveats to remember are:

  1. Packed-sequence support is recipe- and model-family-specific.

  2. Fine-tuning sequence packing should not be assumed to work with every other training feature.

  3. Setting a distinct evaluation CP only reserves compatible data shapes; activating it requires decentralized process groups and caller-managed eval groups. The eval-CP example demonstrates topology plumbing, not a complete real-data recipe; validation sharding and batch math must use the eval DP.

  4. Packed sequences improve efficiency primarily by reducing padding waste, not by replacing long-context parallelism or memory-planning techniques.