nemo_rl.models

Subpackages

• nemo_rl.models.dtensor
  • nemo_rl.models.dtensor.parallelize
• nemo_rl.models.huggingface
  • nemo_rl.models.huggingface.common
• nemo_rl.models.policy
  • nemo_rl.models.policy.utils
  • nemo_rl.models.policy.dtensor_policy_worker
  • nemo_rl.models.policy.hf_policy
  • nemo_rl.models.policy.fsdp1_policy_worker
• nemo_rl.models.generation
  • nemo_rl.models.generation.vllm_backend
  • nemo_rl.models.generation.interfaces
  • nemo_rl.models.generation.vllm
• nemo_rl.models.megatron
  • nemo_rl.models.megatron.common

Submodules

• nemo_rl.models.interfaces