nemo_rl.algorithms

Subpackages

• nemo_rl.algorithms.x_token
  • nemo_rl.algorithms.x_token.token_aligner
  • nemo_rl.algorithms.x_token.loss_utils
  • nemo_rl.algorithms.x_token.utils
• nemo_rl.algorithms.async_utils
  • nemo_rl.algorithms.async_utils.staleness_sampler
  • nemo_rl.algorithms.async_utils.trajectory_collector
  • nemo_rl.algorithms.async_utils.replay_buffer
  • nemo_rl.algorithms.async_utils.interfaces
• nemo_rl.algorithms.single_controller_utils
  • nemo_rl.algorithms.single_controller_utils.config
  • nemo_rl.algorithms.single_controller_utils.rollout_checkpoint
  • nemo_rl.algorithms.single_controller_utils.setup
  • nemo_rl.algorithms.single_controller_utils.utils
  • nemo_rl.algorithms.single_controller_utils.advantage_stage
• nemo_rl.algorithms.loss
  • nemo_rl.algorithms.loss.loss_input
  • nemo_rl.algorithms.loss.clipped_pg
  • nemo_rl.algorithms.loss.draft
  • nemo_rl.algorithms.loss.wrapper
  • nemo_rl.algorithms.loss.utils
  • nemo_rl.algorithms.loss.loss_functions
  • nemo_rl.algorithms.loss.interfaces

Submodules

• nemo_rl.algorithms.grpo_sync
• nemo_rl.algorithms.dpo
• nemo_rl.algorithms.metric_utils
• nemo_rl.algorithms.sft
• nemo_rl.algorithms.single_controller
• nemo_rl.algorithms.grpo
• nemo_rl.algorithms.logits_sampling_utils
• nemo_rl.algorithms.xtoken_off_policy_distillation
• nemo_rl.algorithms.rm
• nemo_rl.algorithms.opd
• nemo_rl.algorithms.reward_functions
• nemo_rl.algorithms.mpo
• nemo_rl.algorithms.distillation
• nemo_rl.algorithms.sft_v2
• nemo_rl.algorithms.advantage_estimator
• nemo_rl.algorithms.utils
• nemo_rl.algorithms.ppo
• nemo_rl.algorithms.advantage_actor