Skip to content

Initial FSDP support in Quantization/Sparsification#10

Draft
hann-wang wants to merge 9 commits into
mainfrom
han/torchtitan_tracking
Draft

Initial FSDP support in Quantization/Sparsification#10
hann-wang wants to merge 9 commits into
mainfrom
han/torchtitan_tracking

Conversation

@hann-wang

@hann-wang hann-wang commented Apr 17, 2026

Copy link
Copy Markdown
Collaborator
  • Track torchtitan upstream changes.
  • selective_ac_option is dropped.
  • torchtitan now defaults to FSDP=1 for single-GPU

This PR enables FSDP in wanda sparsification.

@hann-wang hann-wang changed the title Initial FSDP support Initial FSDP support in Quantization/Sparsification Apr 20, 2026
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