jax.distributed.shutdown# jax.distributed.shutdown()[source]# Shuts down the distributed system. Does nothing if the distributed system is not running.
