jax
93202140 - Make printing work with shard_map after vma has been switched on

Commit

1 year ago

Make printing work with shard_map after vma has been switched on Co-authored-by: Matthew Johnson <mattjj@google.com> PiperOrigin-RevId: 745796269

References

Author

yashk2810

Committer

Google-ML-Automation

Parents