jax
c19e65b7 - fix shard_args logic, closes #1688

Commit
6 years ago
fix shard_args logic, closes #1688 Co-authored-by: Skye Wanderman-Milne <skyewm@google.com>
Author
Committer
Parents
Loading