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.nsys
    • nemo_rl.utils.native_checkpoint
    • nemo_rl.utils.nvml
    • nemo_rl.utils.prefetch_venvs
    • 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.eval_datasets
      • nemo_rl.data.eval_datasets.gpqa
      • nemo_rl.data.eval_datasets.mmlu_pro
      • nemo_rl.data.eval_datasets.mmlu
      • nemo_rl.data.eval_datasets.math
      • nemo_rl.data.eval_datasets.aime2024
      • nemo_rl.data.eval_datasets.local_math_dataset
    • 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.hf_datasets.oai_format_dataset
    • nemo_rl.data.packing
      • nemo_rl.data.packing.algorithms
      • nemo_rl.data.packing.metrics
    • nemo_rl.data.datasets
    • nemo_rl.data.interfaces
    • nemo_rl.data.llm_message_utils
    • nemo_rl.data.processors
  • nemo_rl.distributed
    • nemo_rl.distributed.ray_actor_environment_registry
    • nemo_rl.distributed.model_utils
    • nemo_rl.distributed.named_sharding
    • nemo_rl.distributed.batched_data_dict
    • nemo_rl.distributed.worker_group_utils
    • 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.lm_policy
      • nemo_rl.models.policy.dtensor_policy_worker
      • nemo_rl.models.policy.interfaces
      • nemo_rl.models.policy.megatron_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.converters
        • nemo_rl.models.megatron.converters.common
        • nemo_rl.models.megatron.converters.llama
        • nemo_rl.models.megatron.converters.qwen2
        • nemo_rl.models.megatron.converters.deepseek
        • nemo_rl.models.megatron.converters.qwen3
      • nemo_rl.models.megatron.common
      • nemo_rl.models.megatron.refit_utils
      • nemo_rl.models.megatron.community_import
  • nemo_rl.metrics
    • nemo_rl.metrics.metrics_utils
  • nemo_rl.evals
    • nemo_rl.evals.eval
    • nemo_rl.evals.answer_parsing
  • nemo_rl.package_info

[1]

Created with sphinx-autodoc2