flax
Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names
#5088
Open

Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names #5088

vfdev-5 wants to merge 2 commits into main from use-kernel-metadata-in-gspmd-guide
vfdev-5
review-notebook-app
vfdev-5 vfdev-5 force pushed from 0e021df0 to b1b61b69 49 days ago
vfdev-5 vfdev-5 marked this pull request as ready for review 49 days ago
vfdev-5 vfdev-5 force pushed from b1b61b69 to 841dc02a 44 days ago
vfdev-5 vfdev-5 changed the title Updated flax_gspmd to use kernel_metadata Updated flax_gspmd/bridge_guide to use kernel_metadata, make_mesh and sharding_names 44 days ago
vfdev-5 vfdev-5 force pushed from 841dc02a to 86407e9e 44 days ago
vfdev-5 Updated flax_gspmd to use kernel_metadata
2555dc01
vfdev-5 Updated bridge guide
9023551d
vfdev-5 vfdev-5 force pushed from 86407e9e to 9023551d 42 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone