feat: Support configurable mesh locality and device count in TPU benchmarks#125

linamy85 merged 3 commits into

AI-Hypercomputer:chsAI-Hypercomputer/accelerator-microbenchmarks:chsfrom

simrankaurb:production-mesh-fixsimrankaurb/accelerator-microbenchmarks:production-mesh-fixCopy head branch name to clipboard