API Reference

This page contains auto-generated API reference documentation [1].

• nemo_rl
  • nemo_rl.utils
    • nemo_rl.utils.config
    • nemo_rl.utils.checkpoint
    • nemo_rl.utils.native_checkpoint
    • nemo_rl.utils.nvml
    • nemo_rl.utils.venvs
    • nemo_rl.utils.logger
    • nemo_rl.utils.timer
  • nemo_rl.experience
    • nemo_rl.experience.rollouts
  • nemo_rl.environments
    • nemo_rl.environments.utils
    • nemo_rl.environments.math_environment
    • nemo_rl.environments.interfaces
    • nemo_rl.environments.metrics
  • nemo_rl.data
    • nemo_rl.data.hf_datasets
      • nemo_rl.data.hf_datasets.chat_templates
      • nemo_rl.data.hf_datasets.openmathinstruct2
      • nemo_rl.data.hf_datasets.helpsteer3
      • nemo_rl.data.hf_datasets.squad
      • nemo_rl.data.hf_datasets.prompt_response_dataset
      • nemo_rl.data.hf_datasets.dpo
      • nemo_rl.data.hf_datasets.deepscaler
      • nemo_rl.data.hf_datasets.oasst
    • nemo_rl.data.datasets
    • nemo_rl.data.interfaces
    • nemo_rl.data.llm_message_utils
  • nemo_rl.distributed
    • nemo_rl.distributed.ray_actor_environment_registry
    • nemo_rl.distributed.model_utils
    • nemo_rl.distributed.batched_data_dict
    • nemo_rl.distributed.virtual_cluster
    • nemo_rl.distributed.worker_groups
    • nemo_rl.distributed.collectives
  • nemo_rl.algorithms
    • nemo_rl.algorithms.sft
    • nemo_rl.algorithms.utils
    • nemo_rl.algorithms.loss_functions
    • nemo_rl.algorithms.grpo
    • nemo_rl.algorithms.dpo
    • nemo_rl.algorithms.interfaces
  • nemo_rl.converters
    • nemo_rl.converters.huggingface
      • nemo_rl.converters.huggingface.vllm_export
    • nemo_rl.converters.megatron
      • nemo_rl.converters.megatron.vllm_export
  • nemo_rl.models
    • 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
    • nemo_rl.models.interfaces
  • nemo_rl.metrics
    • nemo_rl.metrics.metrics_utils
  • nemo_rl.evals
    • nemo_rl.evals.eval
  • nemo_rl.package_info

[1]

Created with sphinx-autodoc2