# Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

import metaseq.distributed.rendezvous  # noqa: F401
from .fully_sharded_data_parallel import (
    fsdp_enable_wrap,
    fsdp_wrap,
    FullyShardedDataParallel,
)
from .module_proxy_wrapper import ModuleProxyWrapper


__all__ = [
    "fsdp_enable_wrap",
    "fsdp_wrap",
    "FullyShardedDataParallel",
    "ModuleProxyWrapper",
]
