You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Use jax.device_put_replicated to broadcast to local devices#89
Merged
copybara-service[bot] merged 1 commit intomaingoogle-deepmind/kfac-jax:mainfrom test_499190959google-deepmind/kfac-jax:test_499190959Copy head branch name to clipboardJan 27, 2023
