Skip to content

Latest commit

 

History

History
20 lines (17 loc) · 1.9 KB

File metadata and controls

20 lines (17 loc) · 1.9 KB

Examples

These examples provide concrete examples to leverage vime in your own RL workflow. Some examples are just demonstrative, but most of them are verifiable with a concrete performance score.

Directory Structure

  • eval_multi_task: Example for supporting evaluation multiple tasks with different configs.
  • fully_async: Demonstrates fully asynchronous rollout generation for higher efficiency.
  • geo3k_vlm: Training VLMs on a single-turn reasoning task using GRPO on the GEO3K dataset.
  • geo3k_vlm_multi_turn: VLM multi-turn training on Geo3k dataset.
  • low_precision: Examples of FP8 training and inference for improved throughput and stability.
  • mem_agent: MemAgent long-context RL — chunk-wise memory update, HotpotQA GRPO training, and RULER-HQA evaluation.
  • multi_agent: Example of running multi-agent RL with vime.
  • on_policy_distillation: On-policy distillation (OPD) with an external vLLM teacher or a Megatron-loaded teacher.
  • delta_weight_sync: Non-colocated weight sync that ships only the changed bytes over a shared filesystem (training/inference disaggregation), reloading via the vanilla update_weights_from_disk path.
  • reproducibility: Guides on achieving bitwise experiment reproduction using deterministic modes.
  • retool: Demonstrates the retool functionality for tool-enabled language model generation.
  • search-r1: A minimal reproduction of Search-R1, featuring multi-turn conversation and tool-calling.
  • tau-bench: Multi-turn tool-use agent training in tau-bench environments.
  • train_infer_mismatch_helper: Algorithmic methods for rollout correction (e.g., TIS, MIS).