modelexpress-client 0.7.0

Client library for Model Express gRPC server
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

# 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,
    ObjectStorageConfig,
    StagedWeightVersionShard,
)
from .context import FSDPTrainerContext, MegatronTrainerContext, TrainerEngineContext

__all__ = [
    "FSDPTrainerContext",
    "MegatronTrainerContext",
    "ModelExpressTrainerClient",
    "ModelExpressTrainerConfig",
    "ObjectStorageConfig",
    "StagedWeightVersionShard",
    "TrainerEngineContext",
    "TrainerStagingMode",
    "WeightPayloadFormat",
]