feat(modules): Swin-style PatchMerging with register-row support#122

Open

Dafidofff wants to merge 75 commits into

mainfrom

feat/patch-merging

Commits on May 25, 2026

docs(write/sequence_mixer): add module and class docstrings
David Wessels
committed
docs(write/hyena_nd): add module and class docstrings with math context
David Wessels
committed
docs(review/sequence_mixer): reviewer feedback
David Wessels
committed
docs(review/hyena_nd): reviewer feedback
David Wessels
committed
docs(integrate/sequence_mixer): apply reviewer feedback
David Wessels
committed
docs(write/mixed_fftconv): add module and function docstrings
David Wessels
committed
docs(write/kernels_nd): add module and class docstrings with math context
David Wessels
committed
docs(review/mixed_fftconv): reviewer feedback
David Wessels
committed
docs(review/kernels_nd): reviewer feedback
David Wessels
committed
docs(integrate/mixed_fftconv): apply reviewer feedback
David Wessels
committed
docs(integrate/kernels_nd): apply reviewer feedback
David Wessels
committed
docs(tracker): mark mixed_fftconv, hyena_nd, kernels_nd, sequence_mixer as done
David Wessels
committed
docs(write/patchify): add module and class docstrings
David Wessels
committed
docs(write/residual_block): add module and class docstrings
David Wessels
committed
docs(write/film): add module and class docstrings with math context
David Wessels
committed
docs(review/residual_block): reviewer feedback
David Wessels
committed
docs(review/film): reviewer feedback
David Wessels
committed
docs(review/patchify): reviewer feedback
David Wessels
committed
docs(integrate/residual_block): apply reviewer feedback
David Wessels
committed
docs(integrate/patchify): apply reviewer feedback
David Wessels
committed
docs(write/ckconv_nd): add module and class docstrings with math context
David Wessels
committed
docs(review/ckconv_nd): reviewer feedback
David Wessels
committed
docs(integrate/ckconv_nd): apply reviewer feedback
David Wessels
committed
docs(integrate/film): apply reviewer feedback (from worktree)
David Wessels
committed
docs(tracker): mark ckconv_nd, residual_block, patchify, film as done
David Wessels
committed
docs(write/position_encoding): add module and class docstrings with math context
David Wessels
committed
docs(review/position_encoding): reviewer feedback
David Wessels
committed
docs(write/attention): add module and class docstrings with math context
David Wessels
committed
docs(integrate/position_encoding): apply reviewer feedback
David Wessels
committed
docs(review/attention): reviewer feedback
David Wessels
committed
docs(integrate/attention): apply reviewer feedback
David Wessels
committed
docs(integrate/vit5_residual_block+ckconv_multihead_nd): apply reviewer feedback
David Wessels
committed
docs(tracker): mark attention, position_encoding, ckconv_multihead_nd, vit5_residual_block as done
David Wessels
committed
docs(write/vit5_hyena_adapter): add module and class docstrings
David Wessels
committed
docs(write/condition_mixer): add module and class docstrings
David Wessels
committed
docs(write/mamba_nd): add module and class docstrings with math context
David Wessels
committed
docs(review/condition_mixer): reviewer feedback
David Wessels
committed
docs(integrate/condition_mixer): apply reviewer feedback
David Wessels
committed
docs(write/vit5_attention): add module and class docstrings
David Wessels
committed
docs(write/vit5_hyena_adapter): add module and class docstrings
David Wessels
committed
docs(write/vit5_attention): add module and class docstrings
David Wessels
committed
docs(review/vit5_hyena_adapter): reviewer feedback
David Wessels
committed
docs(review/vit5_attention): reviewer feedback
David Wessels
committed
docs(integrate/vit5_hyena_adapter): apply reviewer feedback
David Wessels
committed
docs(write/mamba_nd): add module and class docstrings with math context
David Wessels
committed
docs(review/mamba_nd): reviewer feedback
David Wessels
committed
docs(integrate/mamba_nd): apply reviewer feedback
David Wessels
committed
docs(integrate/vit5_attention): apply reviewer feedback
David Wessels
committed
docs: merge vit5_attention worktree (write→review→integrate)
David Wessels
committed
docs: merge vit5_hyena_adapter worktree (write→review→integrate)
David Wessels
committed
docs: merge mamba_nd worktree (write→review→integrate)
David Wessels
committed
docs(tracker): mark vit5_attention, vit5_hyena_adapter, condition_mixer, mamba_nd as done
David Wessels
committed

Commits on May 26, 2026

docs(write+integrate/mlp,grn,layer_scale,masks_nd): add module and class docstrings with math context
David Wessels
committed
docs(tracker): mark mlp, grn, layer_scale, masks_nd as done
David Wessels
committed
docs(write+integrate/rms_norm,rms_norm_channel_first,drop_path,causal_conv1d): add module and class docstrings with math context
David Wessels
committed
docs(tracker): mark rms_norm, rms_norm_channel_first, drop_path, causal_conv1d as done
David Wessels
committed
docs(write+integrate/schedulers,distributed_depthwise_conv_nd,general_purpose_resnet,classification_resnet): add module and class docstrings with math/arch context
David Wessels
committed
docs(tracker): mark schedulers, distributed_depthwise_conv_nd, general_purpose_resnet, classification_resnet as done
David Wessels
committed
docs(write+integrate/vit5_classification,a2a_comms,lazy_config,cleanfid,qk_norm,quack_utils): add module/class docstrings
David Wessels
committed
docs(write+integrate/experiments): add module docstrings across run, trainer, default_cfg, lightning_wrappers, datamodules, and utils
David Wessels
committed
docs(tracker): mark vit5_classification, a2a_comms, lazy_config, metrics, utils, testing, and experiments as done
David Wessels
committed
docs(write+integrate/callbacks,ucf101,dali_imagenet_fused): add module docstrings and expand MixupConfig/AugmentConfig
David Wessels
committed
docs(tracker): mark callbacks, ucf101, dali_imagenet_fused as done — documentation complete
David Wessels
committed
docs: add CONVENTIONS.md with docstring style guide and PR enforcement strategy
David Wessels
committed
docs: add docstring CI workflow, extend PR template with doc checklist, expand README docs section
David Wessels
committed
docs: remove stale review artifacts causing Sphinx warnings

David Wessels
and
claude
committed
docs: fix Sphinx cross-ref class names and ViT-5 token layout doc

David Wessels
and
claude
committed
docs(causal_conv1d): fix incorrect Mamba usage note — actual call sites are Hyena short-conv configs

David Wessels
and
claude
committed
docs(drop_path): clarify drop_prob=1.0 is safe — implementation guards div-by-zero

David Wessels
and
claude
committed
fix(ruff): enforce D100 (missing module docstring) — was silently ignored

David Wessels
and
claude
committed
docs: add missing module docstrings to satisfy D100 now that it is enforced

David Wessels
and
claude
committed
feat(modules): add Swin-style PatchMerging with register-row support

authored and
David Wessels
committed
feat(networks+examples): hierarchical ViT-5 classifier and v6 configs

authored and
David Wessels
committed
refactor(v6_hierarchical): move CIFAR-10 configs to cifar10/ subfolder; align ImageNet configs with vit5_hybrid reference

authored and
David Wessels
committed
fix(cifar10): add Lightning datamodule required by v6_hierarchical/cifar10 configs
David Wessels
committed