pytc.vmc.sharding.get_replicated_sharding¶ pytc.vmc.sharding.get_replicated_sharding(mesh)[source]¶ NamedSharding that replicates data on every device. Parameters: mesh (jax.sharding.Mesh)