Skip to content

feat(module): multimodal models infrastructure (DEP-0011) - #43

Open
l1ghtsource wants to merge 1 commit into
d9d-project:mainfrom
l1ghtsource:feat/dep-0011-multimodal-models
Open

feat(module): multimodal models infrastructure (DEP-0011)#43
l1ghtsource wants to merge 1 commit into
d9d-project:mainfrom
l1ghtsource:feat/dep-0011-multimodal-models

Conversation

@l1ghtsource

Copy link
Copy Markdown

Adds multimodality as a first-class entity:

  • MediaSegments + ModalityEncoder trait, MultimodalSequenceInput pipeline input
  • merge_media_embeddings with the empty-media convention
  • vision blocks: patch embedding, interpolated position embedding, 2D rope, packed varlen attention, GeluMLP, VisionBlock, SpatialPatchMerger
  • MultimodalRotaryEmbeddingProvider (interleaved/chunked MRoPE)
  • VarlenSdpaBackend protocol + FA4/FA2/torch implementations
  • dataset helpers: compute_multimodal_position_ids, pad_empty_media
  • fix: probe flash_attn parent before flash_attn.cute in SDPA auto-detection

Verified against HF Qwen3.5-MoE reference modules (vision tower fwd/bwd parity, MRoPE parity, position ids bit-exact vs get_rope_index).

Adds multimodality as a first-class entity:
- MediaSegments + ModalityEncoder trait, MultimodalSequenceInput pipeline input
- merge_media_embeddings with the empty-media convention
- vision blocks: patch embedding, interpolated position embedding, 2D rope,
  packed varlen attention, GeluMLP, VisionBlock, SpatialPatchMerger
- MultimodalRotaryEmbeddingProvider (interleaved/chunked MRoPE)
- VarlenSdpaBackend protocol + FA4/FA2/torch implementations
- dataset helpers: compute_multimodal_position_ids, pad_empty_media
- fix: probe flash_attn parent before flash_attn.cute in SDPA auto-detection

Verified against HF Qwen3.5-MoE reference modules (vision tower fwd/bwd
parity, MRoPE parity, position ids bit-exact vs get_rope_index).
@l1ghtsource
l1ghtsource marked this pull request as draft July 30, 2026 13:15
@l1ghtsource
l1ghtsource marked this pull request as ready for review July 30, 2026 13:15
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