nemo_rl.data.energon

Subpackages

• nemo_rl.data.energon.multimodal
  • nemo_rl.data.energon.multimodal.cookers
    • nemo_rl.data.energon.multimodal.cookers.generic
  • nemo_rl.data.energon.multimodal.task_encoders
    • nemo_rl.data.energon.multimodal.task_encoders.media
    • nemo_rl.data.energon.multimodal.task_encoders.base
    • nemo_rl.data.energon.multimodal.task_encoders.generic_sft
  • nemo_rl.data.energon.multimodal.model_families
  • nemo_rl.data.energon.multimodal.registry
  • nemo_rl.data.energon.multimodal.types
  • nemo_rl.data.energon.multimodal.packing

Submodules

• nemo_rl.data.energon.topology
• nemo_rl.data.energon.config
• nemo_rl.data.energon.sft_dataloader
• nemo_rl.data.energon.sft_worker
• nemo_rl.data.energon.sft_types

Package Contents

Functions

build_energon_sft_loader

Build one rank-aware SFT loader with an optional Energon dependency.

Data

__all__

API

nemo_rl.data.energon.build_energon_sft_loader(**kwargs: Any) → Any

Build one rank-aware SFT loader with an optional Energon dependency.

nemo_rl.data.energon.__all__

[‘EnergonCookerConfig’, ‘EnergonLoaderConfig’, ‘EnergonSourceConfig’, ‘EnergonTaskEncoderConfig’, ‘b…