Wire witness support inside Megatron - #21

Merged

fzyzcjy merged 33 commits into

Jul 9, 2026

Commits on Apr 1, 2026

feat: add generic pre-decoder hook mechanism to GPTModel
fzyzcjy
committed
c5f9de8

Commits on Apr 2, 2026

fix: use direct attribute access for _pre_decoder_hooks
fzyzcjy
committed
77185e1
more
fzyzcjy
committed
153fa7f
refactor: replace pre-decoder hooks with witness_ids forward parameter
fzyzcjy
committed
593ab51
fix: use witness_ids param directly instead of miles_kwargs dict
fzyzcjy
committed
91bbff1
feat: propagate witness_ids through build_schedule_plan path
fzyzcjy
committed
76b54ca
more
fzyzcjy
committed
f12f77b
gitignore
fzyzcjy
committed
4a8e159
feat: add tail_witness support, revert build_schedule_plan changes
fzyzcjy
committed
c83e553
more
fzyzcjy
committed
b7af836
more
fzyzcjy
committed
8225413

Commits on Apr 5, 2026

Commits on Apr 6, 2026

Commits on Jun 5, 2026

Commits on Jun 8, 2026

Require CUDA_DEVICE_MAX_CONNECTIONS != 1 for the fault-tolerant trainer
fzyzcjy
committed
07d03a1

Commits on Jun 9, 2026

Revert "Require CUDA_DEVICE_MAX_CONNECTIONS != 1 for the fault-tolerant trainer"
fzyzcjy
committed
0ed59ff

Commits on Jun 17, 2026

Merge branch 'miles-main' into trainer_ft/dev
fzyzcjy
committed
9910ff5