nemo_rl.models

Subpackages

• nemo_rl.models.megatron
  • nemo_rl.models.megatron.community_import
  • nemo_rl.models.megatron.common
• nemo_rl.models.generation
  • nemo_rl.models.generation.vllm
    • nemo_rl.models.generation.vllm.vllm_generation
    • nemo_rl.models.generation.vllm.vllm_backend
    • nemo_rl.models.generation.vllm.vllm_worker_async
    • nemo_rl.models.generation.vllm.config
    • nemo_rl.models.generation.vllm.utils
    • nemo_rl.models.generation.vllm.vllm_worker
  • nemo_rl.models.generation.fp8
  • nemo_rl.models.generation.interfaces
• nemo_rl.models.huggingface
  • nemo_rl.models.huggingface.common
• nemo_rl.models.policy
  • nemo_rl.models.policy.lm_policy
  • nemo_rl.models.policy.megatron_policy_worker
  • nemo_rl.models.policy.dtensor_policy_worker
  • nemo_rl.models.policy.utils
  • nemo_rl.models.policy.dtensor_policy_worker_v2
  • nemo_rl.models.policy.interfaces
• nemo_rl.models.dtensor
  • nemo_rl.models.dtensor.parallelize