#130 · Primary category: Deep Learning Frameworks
Relax
An Asynchronous Reinforcement Learning Engine for Omni-Modal Post-Training at Scale
Project last updated:08/31/26
GitHub Stars
582
Forks
143
Contributors
45
License
Apache-2.0
Why we included this project
Teams doing reinforcement-learning post-training on large multimodal models will find this one of the few frameworks that handles text, image, video, and audio in a single pipeline. Relax runs as a set of independent Ray Serve services, with Megatron-LM handling training and SGLang handling rollout inference. The rollout, actor, reference, and advantage stages each sit on their own GPU cluster and pass data asynchronously through a streaming transfer queue, which is what lets it scale past the usual synchronous RL loop. It ships with a practical algorithm suite covering PPO, GRPO, GSPO, SAPO, and on-policy distillation, plus an LLM-as-judge reward mode, and it supports agentic multi-turn training with loss masking and elastic rollout scaling for closed-loop agent behaviors. If you already run Ray and Megatron in your stack, this is a serious candidate for production RL post-training rather than a research toy.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.