mesh: add mesh_ring_axis to give a mesh axis physical rings on a torus without wraparound - #4911

mesh: add mesh_ring_axis to give a mesh axis physical rings on a torus without wraparound#4911

systalyze-ai wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

systalyze-ai:upstream-pr/mesh-ring-axissystalyze-ai/maxtext:upstream-pr/mesh-ring-axisCopy head branch name to clipboard