jax
da97ee25 - Stop passing None into jax.NamedSharding in preparation for followup which bans passing None in (in favor of PartitionSpec()

Commit
1 year ago
Stop passing None into jax.NamedSharding in preparation for followup which bans passing None in (in favor of PartitionSpec() PiperOrigin-RevId: 722477002
Author
Parents
Loading