flax
2555dc01 - Updated flax_gspmd to use kernel_metadata

Commit
174 days ago
Updated flax_gspmd to use kernel_metadata
Author
Committer
Parents
Loading