__init__.py

# Copyright (c) 2022-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# See LICENSE for license information.

"""Top level package for numerical debugging."""

try:
    from . import pytorch
    from .pytorch.debug_state import set_weight_tensor_tp_group_reduce
except ImportError as e:
    pass