fix(megatron-fsdp): compute SWiGLU/GDN split in item coordinates for non-DTensor optimizer states - #4423
Open
xuwchen wants to merge 2 commits into
Open
fix(megatron-fsdp): compute SWiGLU/GDN split in item coordinates for non-DTensor optimizer states#4423xuwchen wants to merge 2 commits into
xuwchen wants to merge 2 commits into