[JAX] Keep the routing map format alive and EP multiprocess tests in L2 Jax dist - #3159

[JAX] Keep the routing map format alive and EP multiprocess tests in L2 Jax dist#3159

KshitijLakhani merged 2 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

KshitijLakhani:klakhani/fix/incorrectly-deleted-routing-map-formatKshitijLakhani/TransformerEngine:klakhani/fix/incorrectly-deleted-routing-map-formatCopy head branch name to clipboard