jax
a817f532 - [sharding_in_types] Introduce `auto_mode`, `user_mode`, `auto_mode_ctx` and `user_mode_ctx` as private APIs to make writing auto/user sharding in types code way easier and noise-free.

Commit

1 year ago

[sharding_in_types] Introduce `auto_mode`, `user_mode`, `auto_mode_ctx` and `user_mode_ctx` as **private** APIs to make writing auto/user sharding in types code way easier and noise-free. These can be made public in the future under different names. PiperOrigin-RevId: 714169304

References

#25843 - [sharding_in_types] Introduce `auto_mode`, `user_mode`, `auto_mode_ctx` and `user_mode_ctx` as **private** APIs to make writing auto/user sharding in types code way easier and noise-free.

#213 - CI: 01/27/25 upstream sync

Author

yashk2810

Committer

Google-ML-Automation

Parents

8e86bede

jax a817f532 - [sharding_in_types] Introduce `auto_mode`, `user_mode`, `auto_mode_ctx` and `user_mode_ctx` as **private** APIs to make writing auto/user sharding in types code way easier and noise-free.

jax
a817f532 - [sharding_in_types] Introduce `auto_mode`, `user_mode`, `auto_mode_ctx` and `user_mode_ctx` as private APIs to make writing auto/user sharding in types code way easier and noise-free.