vlmrl
An experimental reinforcement learning gym for vision-language models, written in JAX. It lets you plug in environments and models and train them with PPO.
Pick a small issue you can reproduce, then fix it with your AI coding agent.
Share on XLicense: Apache-2.0
Overview
vlm-gym is an experimental reinforcement learning gym for vision-language models, written in JAX. The idea is to drop in any environment and any model and train with PPO. It ships with environments such as GeoGuessr, NLVR2 and captioning, and a reference Qwen3-VL-4B-Instruct model that can be converted from Hugging Face to JAX. The trainer, rollout engine and evaluation harness live in a core folder, and GeoGuessr training uses a staged curriculum.
Key features
- Pluggable vision environments: GeoGuessr, NLVR2, captioning
- PPO trainer for vision-language models
- Rollout engine and evaluation against a Hugging Face baseline
- Converts Qwen3-VL-4B-Instruct from Hugging Face to JAX
Best for
Suited to researchers experimenting with reinforcement learning on vision-language models in JAX. The README marks its status as experimental.
- Upstream
- sdan/vlm-gym
- Fork on GitHub
- Guo-astro/vlmrl
- Upstream stars
- 150
- Category
- AI agents and LLM tools
- Language
- Python
- License
- Apache-2.0
- Forked
- 2025-10-14
- Sync status
- In syncLast synced 2026-11-09
More in AI agents and LLM tools
/last30days is an AI-agent skill for researching recent discussions across social platforms, video, news, markets, and the web. It brings scattered signals together in a synthesized brief for trend research and preparation.
Forked 2026-10-22Last synced 2026-11-09License: MITAI agents and LLM toolsGitHub
A portable C99 inference engine for running Kimi K3 on a CPU without a GPU or machine-learning framework. It streams the large checkpoint from disk to support inference with limited RAM, for batch text generation or text chat.
Forked 2026-10-21Last synced 2026-11-09License: Apache-2.0AI agents and LLM toolsGitHub
LayerFS gives AI agents writable workspaces over shared history, so they can work in parallel without copying the base. Use it to checkpoint, branch, rewind, and merge agent changes.
Forked 2026-10-21Last synced 2026-11-09License: MITAI agents and LLM toolsGitHub
Magnitude runs open models locally and tunes kernels for your hardware. Use it to connect local models to supported agents or an OpenAI-compatible client.
Forked 2026-10-19Last synced 2026-11-09License: Apache-2.0AI agents and LLM toolsGitHub