Skip to content

[Spyre-Next] Add RowParallelLinear and ColumnParallelLinear(MLP) wrappers - #869

Merged
bohnstingl merged 3 commits into
torch-spyre:mainfrom
nikheal2:wrap_mlp_layer
Apr 2, 2026
Merged

[Spyre-Next] Add RowParallelLinear and ColumnParallelLinear(MLP) wrappers#869
bohnstingl merged 3 commits into
torch-spyre:mainfrom
nikheal2:wrap_mlp_layer