flax
2555dc01 - Updated flax_gspmd to use kernel_metadata

Commit

114 days ago

Updated flax_gspmd to use kernel_metadata

References

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

Author

vfdev-5

vfdev-5

Committer

vfdev-5

vfdev-5

Parents

Loading