axlearn
axlearn copied to clipboard
Pathways: reuse setup_spmd for pathways init
@Ethanlm merged latest main. Do we still want to get this merged?
This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.