Skip to content

feat(transformer): add native manifold hyper-connections - #5936

Closed
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5935from
FDecaYed:codex/hybrid-mhc-native-primitives
Closed

feat(transformer): add native manifold hyper-connections#5936
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5935from
FDecaYed:codex/hybrid-mhc-native-primitives

Conversation

@FDecaYed

@FDecaYed FDecaYed commented Jul 21, 2026

Copy link
Copy Markdown
Contributor

What

Add the model-neutral native PyTorch manifold hyper-connection primitives:

  • expand/reduce residual-stream operations and learned mappings;
  • Sinkhorn projection and initialization controls;
  • selective output-discarding recompute through CheckpointManager;
  • TransformerConfig fields and compatibility validation;
  • focused numerical, gradient, recompute, and serialization tests.

This slice deliberately excludes fused kernels and excludes TransformerBlock, TransformerLayer, GPTModel, HybridModel, pipeline, MTP, and recipe wiring. Those concerns belong to smaller dependent slices.

Stack

Provenance

Testing

  • uv run isort --check-only on all changed Python files
  • uv run ruff check on all changed Python files
  • uv run python -m compileall -q on all changed Python files
  • Focused tests are included; local execution requires the project Torch runtime and is left to stack CI.

Stack follow-up

Add the native PyTorch mHC primitives, configuration, selective-recompute integration, and focused numerical/gradient coverage without model-specific wiring or fused kernels.

Signed-off-by: Deyu Fu <Deyu.Foo@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented Jul 21, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@FDecaYed

Copy link
Copy Markdown
Contributor Author

/ok to test c319ece

Copy link
Copy Markdown
Contributor Author

Closing this draft until Round 1. Its logical successor is P2: complete fused/FP32 mHC with HybridModel and MTP integration, rebuilt after #4531 and P1 merge. It will target main directly rather than remain stacked. Tracking PR: #5795.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant