The APIs for managing the distributed runtime are experimental and potentially subject to future changes.

get_context

nvmath.distributed.get_context() → DistributedContext | None

Return the distributed runtime’s context or None if not initialized.