Infini-AI-Lab
astraflow
Pythonโœจ New

Dataflow-Oriented Reinforcement Learning for (Multi-)Agentic LLMs

Last updated Aug 6, 2026
99
Stars
16
Forks
2
Issues
+1
Stars/day
Attention Score
69
Language breakdown
Python 82.3%
C++ 14.8%
Shell 1.1%
HTML 0.7%
JavaScript 0.7%
CSS 0.3%
โ–ธ Files click to expand
README

AstraFlow โ€” Dataflow-Oriented Reinforcement Learning for (Multi-)Agentic LLMs

arXiv Blog Docs Site License


AstraFlow is a dataflow-oriented reinforcement learning system designed for better flexibility and scalability.

AstraFlow natively supports the following for LLM RL training without any feature-specific system engineering:

  • Fully Async Multi-policy collaborative RL
  • Elastic heterogeneous cross-region rollouts
  • Substitutable rollout and trainer service
  • Composable data algorithms

AstraFlow training a multi-policy workflow on an elastic, heterogeneous, cross-region rollout pool

Fully Async Multi-policy Collaborative RL Training: multiple policies train together, each as an independent trainer with its own data and weight stream.

News

  • [2026/07] AstraFlow v0.1.2 images released โ€” transformers 5.8 / SGLang 0.5.13 stack, new Qwen3.5-4B (Gated-DeltaNet) math recipes, Megatron-image build fix. Both images validated end-to-end on 8ร—H100 (400-step math-RL runs incl. eval). See docker/README.md.
  • [2026/06] New recipe: dynamic recursive agent on TextCraft โ€” a multi-turn agent that recursively spawns sub-agents sharing inventory under a team reward. See the recipe docs.
  • [2026/06] AstraFlow v0.1.1 released โ€” CUDA 13 image, SGLang 0.5.12, Megatron weight-sync training backend, and transformers 5 support. See the project website.
  • [2026/05] AstraFlow v0.1.0 released โ€” first public release of the full system. See the project website.
  • [2026/05] AstraFlow paper is on arXiv.

Getting Started

Recipes

AstraFlow currently supports the following recipes. Check the documentation for more detailed instructions.

| Recipe | Description | |---|---| | math/ | RLVR math reasoning โ€” Qwen3-1.7B / 8B, M2PO, full and delta-weight transfer | | math-multi-agent/ | Actor + verifier collaborative math training | | math-efficient-data/ | Composable data algorithms โ€” GRESO, dynamic sampling, buffer replay | | code/ | Code-generation RL โ€” Qwen3-8B, M2PO | | code-multi-agent/ | Codegen + verifier competitive coding | | search/ | Search-augmented agent training with local retrieval | | alfworld/ | ALFWorld embodied household agent | | webshop/ | WebShop web-navigation shopping agent |

Roadmap

Near-term focus:
  • [ ] Offline cluster training โ€” Support training on offline clusters without internet access.
  • [ ] All-in-one launcher โ€” A launcher helper that streamlines bringing up the AstraFlow, RaaS, and trainer services.
  • [ ] MoE model support โ€” Extend the training backends to Mixture-of-Experts models.
  • [ ] Terminal-Bench training โ€” Add a recipe for training agents on Terminal-Bench.
  • [ ] Megatron backend โ€” Add Megatron-LM as a training backend.
  • [ ] vLLM rollout engine โ€” Support vLLM alongside SGLang as a rollout engine.

Citation

If you find AstraFlow useful in your research, please cite:
@article{zheng2026astraflow,
  title   = {AstraFlow: Dataflow-Oriented Reinforcement Learning for Agentic LLMs},
  author  = {Zheng, Haizhong and Di, Yizhuo and Wang, Jiahui and Jin, Shuowei and
             Liu, Xueshen and Wu, Yongji and Mao, Z. Morley and Stoica, Ion and
             Zhao, Jiawei and Chen, Beidi},
  journal = {arXiv preprint arXiv:2605.15565},
  year    = {2026}
}

Acknowledgment

We learned the design and reused code from the following projects: AReaL, verl, AgentBench, ASearcher, and M2PO.
๐Ÿ”— More in this category

ยฉ 2026 GitRepoTrend ยท Infini-AI-Lab/astraflow ยท Updated daily from GitHub