Python get_shard_indices - 4 examples found. These are the top rated real world Python examples of jax.experimental.global_device_array.get_shard_indices extracted from open source projects. You can rate examples to help us improve the quality of examples.