Test include stride_per_key to KJT's flatten and unflatten#2903

Open

jd7-tr wants to merge 1 commit intopytorch:mainfrom

jd7-tr:export-D73051959

Commits on May 1, 2025

Include stride_per_key_per_rank in KJT's PyTree flatten/unflatten logic (#2903)

jd7-tr
authored and
facebook-github-bot
committed