jax
97b1faac - Fixes the random key sharding in shard_map.

Commit

1 year ago

Fixes the random key sharding in shard_map.

References

#25692 - Fixes incorrect sharding lowering for key<fry> in shard_map partial-auto

#193 - CI: 01/06/25 upstream sync

Author

yliu120

yliu120

Committer

yliu120

yliu120

Parents

Loading