modelexpress-client 0.6.0

# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

"""Trainer-side ModelExpress RL integrations."""

from .adapter import TrainerStagingMode, WeightPayloadFormat
from .client import (
    ModelExpressTrainerClient,
    ModelExpressTrainerConfig,
    StagedWeightVersionShard,
)

__all__ = [
    "ModelExpressTrainerClient",
    "ModelExpressTrainerConfig",
    "StagedWeightVersionShard",
    "TrainerStagingMode",
    "WeightPayloadFormat",
]