diff --git a/.github/ISSUE_TEMPLATE/config.yml b/.github/ISSUE_TEMPLATE/config.yml index 34faf1138..e997cf03b 100644 --- a/.github/ISSUE_TEMPLATE/config.yml +++ b/.github/ISSUE_TEMPLATE/config.yml @@ -1,8 +1,8 @@ blank_issues_enabled: true contact_links: - name: Documentation - url: https://redai-infra.github.io/Relax + url: https://redai-studio.github.io/Relax about: Check the documentation for guides and API reference - name: Questions & Discussions - url: https://github.com/redai-infra/Relax/discussions + url: https://github.com/redai-studio/Relax/discussions about: Ask questions and discuss ideas with the community diff --git a/.github/workflows/deploy-docs.yml b/.github/workflows/deploy-docs.yml index 9d8719240..bd797817a 100644 --- a/.github/workflows/deploy-docs.yml +++ b/.github/workflows/deploy-docs.yml @@ -22,7 +22,7 @@ concurrency: jobs: build: runs-on: ubuntu-latest - if: github.repository == 'redai-infra/Relax' + if: github.repository == 'redai-studio/Relax' steps: - name: Checkout uses: actions/checkout@v4 @@ -61,7 +61,7 @@ jobs: deploy: needs: build runs-on: ubuntu-latest - if: github.repository == 'redai-infra/Relax' + if: github.repository == 'redai-studio/Relax' environment: name: github-pages url: ${{ steps.deployment.outputs.page_url }} diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 062d4ebd3..b24745f0d 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -42,7 +42,7 @@ This project follows a standard code of conduct. Please be respectful, inclusive ```bash # Clone the repository -git clone https://github.com/redai-infra/Relax.git +git clone https://github.com/redai-studio/Relax.git cd Relax # Create virtual environment (recommended) @@ -155,7 +155,7 @@ feat(rollout): add streaming data consumption for async mode ## Reporting Bugs -Use the [Bug Report template](https://github.com/redai-infra/Relax/issues/new?template=bug_report.md) and include: +Use the [Bug Report template](https://github.com/redai-studio/Relax/issues/new?template=bug_report.md) and include: - **Environment** — OS, Python version, CUDA version, GPU type - **Steps to Reproduce** — Minimal commands to trigger the bug @@ -165,7 +165,7 @@ Use the [Bug Report template](https://github.com/redai-infra/Relax/issues/new?te ## Requesting Features -Use the [Feature Request template](https://github.com/redai-infra/Relax/issues/new?template=feature_request.md) and include: +Use the [Feature Request template](https://github.com/redai-studio/Relax/issues/new?template=feature_request.md) and include: - **Problem Statement** — What problem does this solve? - **Proposed Solution** — How should it work? diff --git a/README.md b/README.md index 03c93a2e8..79d644df8 100644 --- a/README.md +++ b/README.md @@ -16,13 +16,13 @@ arXiv - + Documentation - + WeChat QR - + Docker Image

@@ -34,7 +34,7 @@ ______________________________________________________________________ -**Relax** (**R**einforcement **E**ngine **L**everaging **A**gentic **X**-modality) is a high-performance reinforcement learning post-training framework open-sourced by the Xiaohongshu AI Infra Team for multimodal large language models. Built on Ray Serve with a service-oriented architecture, Relax uses Megatron-LM as the training backend and SGLang as the inference engine. Through the [TransferQueue](https://github.com/redai-infra/TransferQueue) data transfer system, it achieves complete decoupling of training and inference, supporting end-to-end multimodal RL training from text to images, videos, and audio. +**Relax** (**R**einforcement **E**ngine **L**everaging **A**gentic **X**-modality) is a high-performance reinforcement learning post-training framework open-sourced by the Xiaohongshu AI Infra Team for multimodal large language models. Built on Ray Serve with a service-oriented architecture, Relax uses Megatron-LM as the training backend and SGLang as the inference engine. Through the [TransferQueue](https://github.com/redai-studio/TransferQueue) data transfer system, it achieves complete decoupling of training and inference, supporting end-to-end multimodal RL training from text to images, videos, and audio. ______________________________________________________________________ @@ -129,15 +129,15 @@ The recommended way to run Relax is via the official Docker image, which ships w ```bash # Pull the official image -docker pull ghcr.io/redai-infra/relaxrl:latest +docker pull ghcr.io/redai-studio/relaxrl:latest # Launch a container with GPUs, shared memory, and your workspace mounted docker run -it --gpus all --ipc=host --network=host \ -v /path/to/your/workspace:/root \ - ghcr.io/redai-infra/relaxrl:latest bash + ghcr.io/redai-studio/relaxrl:latest bash # Inside the container -git clone https://github.com/redai-infra/Relax.git /root/Relax +git clone https://github.com/redai-studio/Relax.git /root/Relax cd /root/Relax && pip install -e . ``` @@ -246,7 +246,7 @@ ______________________________________________________________________ ## 📚 Documentation -Full bilingual documentation is available at **[redai-infra.github.io/Relax](https://redai-infra.github.io/Relax)**. +Full bilingual documentation is available at **[redai-studio.github.io/Relax](https://redai-studio.github.io/Relax)**. ______________________________________________________________________ diff --git a/README_zh.md b/README_zh.md index af13a4c36..faf13e888 100644 --- a/README_zh.md +++ b/README_zh.md @@ -16,13 +16,13 @@ arXiv - + Documentation - + WeChat QR - + Docker Image

@@ -34,7 +34,7 @@ ______________________________________________________________________ -**Relax**（**R**einforcement **E**ngine **L**everaging **A**gentic **X**-modality）是小红书 AI 平台开源的、面向多模态大模型的高性能强化学习后训练框架。Relax 基于 Ray Serve 构建面向服务的架构，以 Megatron-LM 为训练后端、SGLang 为推理引擎，通过 [TransferQueue](https://github.com/redai-infra/TransferQueue) 数据传输系统实现训练与推理的完全解耦，支持从文本到图像、视频、音频的全模态强化学习训练。 +**Relax**（**R**einforcement **E**ngine **L**everaging **A**gentic **X**-modality）是小红书 AI 平台开源的、面向多模态大模型的高性能强化学习后训练框架。Relax 基于 Ray Serve 构建面向服务的架构，以 Megatron-LM 为训练后端、SGLang 为推理引擎，通过 [TransferQueue](https://github.com/redai-studio/TransferQueue) 数据传输系统实现训练与推理的完全解耦，支持从文本到图像、视频、音频的全模态强化学习训练。 ______________________________________________________________________ @@ -129,15 +129,15 @@ ______________________________________________________________________ ```bash # 拉取官方镜像 -docker pull ghcr.io/redai-infra/relaxrl:latest +docker pull ghcr.io/redai-studio/relaxrl:latest # 启动容器，挂载 GPU、共享内存与工作目录 docker run -it --gpus all --ipc=host --network=host \ -v /path/to/your/workspace:/root \ - ghcr.io/redai-infra/relaxrl:latest bash + ghcr.io/redai-studio/relaxrl:latest bash # 容器内克隆仓库并安装 -git clone https://github.com/redai-infra/Relax.git /root/Relax +git clone https://github.com/redai-studio/Relax.git /root/Relax cd /root/Relax && pip install -e . ``` @@ -246,7 +246,7 @@ ______________________________________________________________________ ## 📚 文档 -完整的双语文档请访问 **[redai-infra.github.io/Relax](https://redai-infra.github.io/Relax)**。 +完整的双语文档请访问 **[redai-studio.github.io/Relax](https://redai-studio.github.io/Relax)**。 ______________________________________________________________________ diff --git a/docker/Dockerfile b/docker/Dockerfile index c473e89b0..70e98cd9f 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -68,7 +68,7 @@ RUN MAX_JOBS=64 \ rm -rf /opt/flash-attention/ RUN pip -v install --no-cache-dir --no-build-isolation "transformer_engine[pytorch]==2.14.1" && \ - TMS_CUDA_MAJOR=$(python -c 'import torch; print(torch.version.cuda.split(".")[0])') pip install git+https://github.com/redai-infra/torch_memory_saver.git@afc13785c50119048e2dd8ac497cc9e29ec75bd4 --no-cache-dir --force-reinstall && \ + TMS_CUDA_MAJOR=$(python -c 'import torch; print(torch.version.cuda.split(".")[0])') pip install git+https://github.com/redai-studio/torch_memory_saver.git@afc13785c50119048e2dd8ac497cc9e29ec75bd4 --no-cache-dir --force-reinstall && \ pip install nvidia-modelopt[torch]>=0.37.0 --no-build-isolation --no-cache-dir && \ pip install "numpy<2" nvidia-cudnn-cu12==9.16.0.29 --no-cache-dir && \ NVCC_APPEND_FLAGS="--threads 32" \ @@ -116,7 +116,7 @@ COPY requirements.txt /tmp/requirements.txt RUN pip install --ignore-installed PyJWT && \ pip install -r /tmp/requirements.txt --no-cache-dir && \ pip install --no-cache-dir "compressed_tensors>=0.13.0" tensordict==0.10.0 pyvers==0.1.0 'nvidia-modelopt[hf]==0.44.0' --no-deps && \ - pip install "transferqueue @ git+https://github.com/redai-infra/TransferQueue.git@58054a33834aadbcf76aacd6b1e32e25c030f2c9" --no-deps + pip install "transferqueue @ git+https://github.com/redai-studio/TransferQueue.git@58054a33834aadbcf76aacd6b1e32e25c030f2c9" --no-deps # sgl-router: override the official wheel (pulled by requirements.txt above) with # slime's r3-capable fork. The official sglang-router drops the routed_experts diff --git a/docker/Dockerfile.npu b/docker/Dockerfile.npu index 0eee80263..7caa97acb 100644 --- a/docker/Dockerfile.npu +++ b/docker/Dockerfile.npu @@ -79,7 +79,7 @@ RUN pip_install() { pip install "$@" || pip install --index-url https://pypi.org cd /root && rm -rf /root/pytorch && \ pip_install triton-ascend==3.2.0 && \ pip_install tensordict==0.10.0 pyvers==0.1.0 --no-deps -RUN git clone https://github.com/redai-infra/TransferQueue.git /root/TransferQueue && \ +RUN git clone https://github.com/redai-studio/TransferQueue.git /root/TransferQueue && \ cd /root/TransferQueue && \ git checkout 58054a33834aadbcf76aacd6b1e32e25c030f2c9 && \ pip install /root/TransferQueue --no-deps --no-build-isolation && \ diff --git a/docs/.vitepress/config.mts b/docs/.vitepress/config.mts index b9761c907..5b6a51acd 100644 --- a/docs/.vitepress/config.mts +++ b/docs/.vitepress/config.mts @@ -229,7 +229,7 @@ export default defineConfig({ { text: 'Resources', items: [ - { text: 'GitHub', link: 'https://github.com/redai-infra/Relax' }, + { text: 'GitHub', link: 'https://github.com/redai-studio/Relax' }, { text: 'Paper', link: 'https://arxiv.org/abs/2604.11554' } ] } @@ -345,7 +345,7 @@ export default defineConfig({ { text: '资源', items: [ - { text: 'GitHub', link: 'https://github.com/redai-infra/Relax' }, + { text: 'GitHub', link: 'https://github.com/redai-studio/Relax' }, { text: '论文', link: 'https://arxiv.org/abs/2604.11554' } ] } @@ -474,7 +474,7 @@ export default defineConfig({ themeConfig: { logo: '/rednote-logo.png', socialLinks: [ - { icon: 'github', link: 'https://github.com/redai-infra/Relax' } + { icon: 'github', link: 'https://github.com/redai-studio/Relax' } ], search: { provider: 'local' diff --git a/docs/.vitepress/theme/CallToAction.vue b/docs/.vitepress/theme/CallToAction.vue index ed957b6d7..674f503e5 100644 --- a/docs/.vitepress/theme/CallToAction.vue +++ b/docs/.vitepress/theme/CallToAction.vue @@ -26,8 +26,8 @@ const subtitle = computed(() => const primaryLabel = computed(() => isZh.value ? '参与贡献' : 'Contribute Now') const secondaryLabel = computed(() => isZh.value ? '讨论区' : 'Discussion') -const primaryLink = 'https://github.com/redai-infra/Relax/blob/main/CONTRIBUTING.md' -const secondaryLink = 'https://github.com/redai-infra/Relax/discussions' +const primaryLink = 'https://github.com/redai-studio/Relax/blob/main/CONTRIBUTING.md' +const secondaryLink = 'https://github.com/redai-studio/Relax/discussions'