Update the documentation for `shardformer` before merging into the main branch.
Update the documentation for
shardformerbefore merging into the main branch.