Skip to content

add high precision init weights to fully_shard example#2785

Draft
pstjohn wants to merge 1 commit intoNVIDIA:mainfrom
pstjohn:worktree-pstjohn/clean-up-example
Draft

add high precision init weights to fully_shard example#2785
pstjohn wants to merge 1 commit intoNVIDIA:mainfrom
pstjohn:worktree-pstjohn/clean-up-example

Conversation

@pstjohn
Copy link
Contributor

@pstjohn pstjohn commented Mar 20, 2026

WIP, would like to also add tests around preserve_high_precision_init_val

Signed-off-by: Peter St. John <pstjohn@nvidia.com>
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