Move ifrt based reshard out of experimental. Leaving intermediate resharding and sidechannel resharding in experimental.

lukebaumann · The tunix Authors · commit 52a611cdf0c3 · 2026-02-26T12:54:51.000-08:00
PiperOrigin-RevId: 875850174
diff --git a/tunix/rl/reshard.py b/tunix/rl/reshard.py
@@ -366,7 +366,7 @@ def reshard_fn(
           )
 
       # TODO(b/476149699): Migrate to new API once it's verified.
-      return experimental_reshard._reshard_with_sidechannel(
+      return experimental_reshard.reshard(
           x,
           sharding,
           donate=donate,

Original file line number	Diff line number	Diff line change
`@@ -366,7 +366,7 @@ def reshard_fn(`
`366`	`366`	`)`
`367`	`367`
`368`	`368`	`# TODO(b/476149699): Migrate to new API once it's verified.`
`369`		`- return experimental_reshard._reshard_with_sidechannel(`
	`369`	`+ return experimental_reshard.reshard(`
`370`	`370`	`x,`
`371`	`371`	`sharding,`
`372`	`372`	`donate=donate,`