Skip to content

Add versioned trainer graphs and logical-rank APIs - #925

Draft
bradhilton wants to merge 147 commits into
mainfrom
stark/trainer-v1-20260917
Draft

bradhilton wants to merge 147 commits into
mainfrom
stark/trainer-v1-20260917

Conversation

@bradhilton

@bradhilton bradhilton commented Sep 18, 2026 •

Copy link
Copy Markdown
Collaborator

Adds planner-managed backward-state retention, CPU offload and replay, bounded gradient staleness, managed output tensors, and checkpoint-owned live parameters/modules. Rank-zero and logical-DP views support the paired Caladan client API.

At 2e1281ac, CPU CI passes 1,519 Megatron and 3,925 unit tests; GPU CI passes 596 tests plus CP=2/TP=2 correctness checks. Skips remain explicit in the original runs. Focused snapshot, ownership and test-consolidation checks also pass.

Earlier Qwen3-8B throughput deltas remain −12.1% at 4×1K and −4.5% at 4×4K. Draft for API review, validation and further size reductions.

@bradhilton
bradhilton had a problem deploying to trainer-rank-gpu-validation September 18, 2026 00:36 — with GitHub Actions Failure
@bradhilton
bradhilton had a problem deploying to trainer-rank-gpu-validation September 18, 2026 01:04 — with GitHub Actions Error
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 18, 2026 01:24 — with GitHub Actions Active
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 19, 2026 21:25 — with GitHub Actions Active
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 20, 2026 02:55 — with GitHub Actions Active
@bradhilton
bradhilton had a problem deploying to trainer-rank-gpu-validation September 29, 2026 00:01 — with GitHub Actions Failure
@bradhilton
bradhilton had a problem deploying to trainer-rank-gpu-validation September 29, 2026 00:46 — with GitHub Actions Failure
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 29, 2026 01:22 — with GitHub Actions Active
@bradhilton
bradhilton had a problem deploying to trainer-rank-gpu-validation September 29, 2026 06:19 — with GitHub Actions Error
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 29, 2026 06:39 — with GitHub Actions Active
@bradhilton
bradhilton deployed to trainer-rank-gpu-validation September 29, 2026 07:13 — with GitHub Actions Active

This branch was successfully deployed

1 active deployment
trainer-rank-gpu-validation — 2e1281ac Deployed Sep 29, 2026 by bradhilton via Run on 2x H200 #977
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant