nemo_rl.data_plane

NeMo-RL data-plane package.

The public surface is intentionally tiny: an ABC, a meta dataclass, a config TypedDict, and a factory. Everything else is an implementation detail of a specific adapter.

Subpackages

• nemo_rl.data_plane.adapters
  • nemo_rl.data_plane.adapters.local
  • nemo_rl.data_plane.adapters.noop
  • nemo_rl.data_plane.adapters.transfer_queue
  • nemo_rl.data_plane.adapters.tq_mooncake_checkpoint
  • nemo_rl.data_plane.adapters.transfer_queue_env

Submodules

• nemo_rl.data_plane.driver_mixin
• nemo_rl.data_plane.schema
• nemo_rl.data_plane.interfaces
• nemo_rl.data_plane.tq_token_sink
• nemo_rl.data_plane.codec
• nemo_rl.data_plane.observability
• nemo_rl.data_plane.preshard
• nemo_rl.data_plane.column_io
• nemo_rl.data_plane.worker_mixin
• nemo_rl.data_plane.async_utils
• nemo_rl.data_plane.factory

Package Contents

Data

__all__

API

nemo_rl.data_plane.__all__

[‘DATA_PLANE_CHECKPOINT_SCHEMA_VERSION’, ‘DataPlaneClient’, ‘DataPlaneConfig’, ‘KVBatchMeta’, ‘Metri…