[0z5a]

Summary

Brings 0z5a-branch up to main (d7e40fe) with one merge commit (c1027a5f, signed off), so later work can be based on main and reviewed incrementally against this branch.

The merge adds the three main commits that 0z5a-branch lacked:

• [Bugfix][Hardware][AMD] Fix colocate weight sync on ROCm and add a ROCm 7.14 Dockerfile #418 ROCm colocate weight-sync fix and ROCm 7.14 Dockerfile;
• [Sync] Mirror Slime v0.4.0 training stack and release updates #441 Slime v0.4.0 training-stack sync, including the Megatron pin 4b4acac (MCore 0.19) and its MTP module-name mapping;
• [CI][ROCm] Add ROCm CI: Dockerfile, Buildkite pipeline, and test variants #474 ROCm CI.

How conflicts were resolved

main is the reference. Every conflicting hunk takes main's version (git merge -X theirs origin/main). Changes that exist only on 0z5a-branch and do not conflict are kept:

• [Feature][Training] Collect versioned target features for external draft #454 versioned target-feature collection for external drafts;
• [Training][Rollout] Integrate Ouro shared layers and policy updates #459 Ouro shared layers and policy updates;
• [Maintenance] Consolidate rollout, checkpoint, topology fixes and colocate docs #460 consolidation of rollout, checkpoint and topology fixes and colocate docs.

docker/ is identical to main.

Differences from main after this merge

45 files, +2942 / −48, all from #454, #459 and #460:

Area	Files
Draft-feature collection (#454)	vime/utils/draft_feature_contract.py, vime/backends/megatron_utils/draft_feature_collector.py, actor/arguments hooks, docs advanced/draft-feature-collection.md, tests/test_draft_feature_*.py, tests/integration/run_draft_feature_ouro.py
Ouro (#459)	vime_plugins/ouro/, examples/ouro/, Ouro loader and exporter entries in hf_to_megatron / megatron_to_hf
Consolidation (#460)	vime/rollout/vllm_rollout.py, vime/utils/arguments.py, vime/utils/reward_normalization.py, vime/utils/tensor_backper.py, vime/observability/*, examples/mem_agent/_common.sh, docs get_started/usage.md, tests/test_server_idle_contract.py, tests/test_mem_agent_nvlink_detect.py, tests/integration/test_native_adam_resume.py

CI, as differences from main

I ran the 44 CPU test files listed in .buildkite/pipeline.yml on main and on this branch:

• 43 files: same result as main.
• tests/test_megatron_argument_validation.py: on main, 71 passed. Here, 87 passed and 22 failed. All 22 failing tests exist only on 0z5a-branch:
  • test_colocate_offload_resolution (18 cases, [Maintenance] Consolidate rollout, checkpoint, topology fixes and colocate docs #460);
  • test_collect_only_supported_arguments and test_collect_only_rejects_before_side_effects (4 cases, [Feature][Training] Collect versioned target features for external draft #454).
  • They check argument handling for which main's logic won the conflict in vime/utils/arguments.py: colocate offload defaults, and validation of the collect-only mode.
  • So [Maintenance] Consolidate rollout, checkpoint, topology fixes and colocate docs #460's offload defaults and [Feature][Training] Collect versioned target features for external draft #454's collect-only validation are not active after this merge. Re-applying them on top of main is follow-up work.
• DCO: d7e40fe8 ([CI][ROCm] Add ROCm CI: Dockerfile, Buildkite pipeline, and test variants #474) has no Signed-off-by on main itself; this PR does not change it. The merge commit is signed off.

GPU tests were not run for this merge.

AI assistance

The merge, the checks and this description were made with Claude Code (AI-assisted).

🤖 Generated with Claude Code

[read-the-docs-community]

Documentation build overview

📚 vime | 🛠️ Build #35253310 | 📁 Comparing c1027a5 against latest (d7e40fe)

  🔍 Preview build  

60 files changed · + 2 added · ± 58 modified

+ Added

• advanced/draft-feature-collection.html
• _examples_synced/ouro/README.html

± Modified

• genindex.html
• library.html
• search.html
• advanced/arch-support-beyond-megatron.html
• advanced/delta-weight-sync.html
• advanced/external-rollout-engines.html
• advanced/fault-tolerance.html
• advanced/low-precision.html
• advanced/megatron-config.html
• advanced/observability.html
• and 48 more...

[gemini-code-assist]

Code Review

This pull request updates the Vime repository by introducing an interactive experiment builder lab to the documentation, restructuring the docs directory, and updating the Docker and Buildkite CI configurations to support new features like speculative decoding, fault tolerance, and straw transport. Feedback on the changes includes installing the missing patch utility in docker/Dockerfile.rocm to prevent build failures and updating a legacy [slime-docs] prefix to [vime-docs] in docs/build.sh for consistency.

[gemini-code-assist]

The patch utility is used later in the Dockerfile (lines 126-129) to apply various vLLM patches, but it is not installed in this stage. Since the base image vllm/vllm-openai-rocm:nightly may not have patch pre-installed, please add patch to the apt-get install command to prevent potential build failures.

RUN apt-get update && apt-get install -y \
    rsync dnsutils iproute2 prometheus patch && \
    rm -rf /var/lib/apt/lists/*

[gemini-code-assist]

The error message refers to [slime-docs], which is the old name of the project. Since the project has been renamed to Vime, please update this prefix to [vime-docs] to maintain consistency with other build scripts (e.g., docs/build_all.sh uses [vime-docs]).

Suggested change

	if [ -z "$SPHINX_BUILD" ]; then
	cat >&2 <<'EOF'
	[slime-docs] sphinx-build was not found. From the repository root, run:
	python3 -m venv .venv-docs
	.venv-docs/bin/python -m pip install -r docs/requirements.txt
	Then retry: bash docs/build_all.sh
	EOF
	exit 1
	fi
	if [ -z "$SPHINX_BUILD" ]; then
	cat >&2 <<'EOF'
	[vime-docs] sphinx-build was not found. From the repository root, run:
	python3 -m venv .venv-docs
	.venv-docs/bin/python -m pip install -r docs/requirements.txt
	Then retry: bash docs/build_all.sh
	EOF
	exit 1
	fi