jax
515f81bf - Shut down `PreemptionSyncManager` when `jax.distributed.shutdown()` is called.
Commit
View On GitHub
Commit
1 year ago
Shut down `PreemptionSyncManager` when `jax.distributed.shutdown()` is called. PiperOrigin-RevId: 756398914
Author
mwhittaker mwhittaker
Parents
6c66e938
FAQ Terms Privacy Refunds Impressum
Loading