This is your work, valued
Researcher at Thinking Machines Lab | PhD Student at UC Berkeley (EECS) | UC Berkeley 22' (CS) | Research interests: Improved Reasoning on LLMs/VLMs through RL
LLM-groundedDiffusion. LLM-grounded Diffusion: Enhancing Prompt Understanding of Text-to-Image Diffusion Models with Large Language Models (LLM-grounded Diffusion: LMD, TMLR 2024)
★ 481stable-diffusion-xl-demo. A gradio web UI demo for Stable Diffusion XL 1.0, with refiner and MultiGPU support
★ 279AnimeGAN.js. AnimeGAN.js: Photo Animation for Everyone
★ 229LLM-groundedVideoDiffusion. [ICLR 2024] LLM-grounded Video Diffusion Models (LVD): official implementation for the LVD paper
★ 172CrossMAE. Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders
★ 135UnsupervisedSelectiveLabeling. [ECCV 2022] Official Implementation for Unsupervised Selective Labeling for More Effective Semi-Supervised Learning
★ 64igligen. Improved Implementation for Training GLIGEN: Open-Set Grounded Text-to-Image Generation
★ 46RCF-UnsupVideoSeg. [CVPR 2023] Segmenting objects in videos without human annotations 🤯: Official implementation for Bootstrapping Objectness from Videos by Relaxed Common Fate and Visual Grouping
★ 40uARM.wasm. An ARM emulator ported to WebAssembly and running Linux in browsers.
★ 35VAI-ReinforcementLearning. [CVPR 2021] Official Implementation of VAI: Unsupervised Visual Attention and Invariance for Reinforcement Learning
★ 27RIDE-Swin-LongTailRecognition. Swin-Transformer-based version of RIDE (ICLR 2021 Spotlight)
★ 12lotus. An interactive utility that breaks the interdependency between deep learning and coding
★ 11longbox. An iOS App which can let you run Windows or Linux on your iOS devices without jailbreak, based on ported version of bochs.
★ 9cubie-5110. Nokia 5110 driver using GPIO on Cubieboard, Cubietruck, and other A10,A13,A20 boards
★ 6beauticam. 一个萌萌的小创意——给你的脸增加猫耳朵(纯web前端, Vue.js)
★ 6spi_flash. spi flash driver in linux user space using gpio.
★ 5colorQuest. An online game based on HTML5/JavaScript.
★ 2jsdos. A emulator which is written in JavaScript to run DOS system.
★ 2uARMSim. uARM Simulation Version
★ 1LiuXueEatChicken. 留学吃鸡知识竞赛H5源码
★ 1Query.jl. Query almost anything in julia
★ 1PARL-Sample. Deep reinforcement learning using baidu PARL(maze,flappy bird and so on)
★ 1pytorch-template. PyTorch deep learning projects made easy.
★ 1sglang. SGLang is a fast serving framework for large language models and vision language models.
★ 1jekyll-sass-converter. A Sass converter for Jekyll.
★ 1agents-last-exam. Agents' Last Exam
★ 838PostTrainBench. Measuring how well CLI agents like Claude Code or Codex CLI can post-train base LLMs on a single H100 GPU in 10 hours
★ 432first-steps-toward-automated-ai-research. Research artifacts from Recursive's automated AI research system
★ 162hypergrep. Recursively search directories for a regex pattern
★ 245hil-bench. Python
★ 26deep-swe. Measuring frontier coding agents on original, long-horizon engineering tasks
★ 1.1kOpenClaw-RL. OpenClaw-RL: Train any agent simply by talking
★ 5.6kdesign.md. A format specification for describing a visual identity to coding agents. DESIGN.md gives agents a persistent, structured understanding of a design system.
★ 26kawesome-design-md. A collection of DESIGN.md files analysis by popular brand design systems. Drop one into your project and let coding agents generate a matching UI.
★ 101kAutoGaze. AutoGaze automatically removes redundant patches in a video, reducing #tokens in ViT/MLLM by 4x-100x.
★ 294Stirrup. The lightweight framework for building agents
★ 493OpenResearcher. OpenResearcher: A Fully Open Pipeline for Long-Horizon Deep Research Trajectory Synthesis
★ 800fzf. :cherry_blossom: A command-line fuzzy finder
★ 82kwhisper.cpp. Port of OpenAI's Whisper model in C/C++
★ 52knvm. Node Version Manager - POSIX-compliant bash script to manage multiple active node.js versions
★ 94kviu. Terminal image viewer with native support for iTerm and Kitty
★ 3.2kslime. slime is an LLM post-training framework for RL Scaling.
★ 7.4kopenui. The Open Standard for Generative UI
★ 7.9kopen-webui. User-friendly AI Interface (Supports Ollama, OpenAI API, ...)
★ 145kVisGym. Official Repository of VisGym: Diverse, Customizable, Scalable Environments for Multimodal Agents
★ 114VPBench. Implementation for VPBench proposed in paper Visually Prompted Benchmarks Are Surprisingly Fragile
★ 11DeepResearch. Tongyi Deep Research, the Leading Open-source Deep Research Agent
★ 20ksimple-evals. Python
★ 4.6kghstack. Submit stacked diffs to GitHub on the command line
★ 999tau-bench. Code and Data for Tau-Bench
★ 1.3ktorchtitan. A PyTorch native platform for training generative AI models
★ 5.5kthreadweaver. The implementation for ThreadWeaver Adaptive Threading for Efficient Parallel Reasoning in Language Models
★ 65rate-evals. A comprehensive evaluation pipeline for Vision-Language Models on medical imaging tasks, with built-in support for multi-GPU processing, real-time progress tracking, and disease finding classification.
★ 17StringSight. Automatically Analyze your Model Traces
★ 45Native-Parallel-Reasoner. [ICML 2026] Reasoning in Parallelism via Self-Distilled RL
★ 113K2-Vendor-Verifier. Verify Precision of all Kimi K2 API Vendor
★ 578tinker-cookbook. Post-training with Tinker
★ 3.6kMMMU. This repo contains evaluation code for the paper "MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGI"
★ 587nanochat. The best ChatGPT that $100 can buy.
★ 56knanoGPT. The simplest, fastest repository for training/finetuning medium-sized GPTs.
★ 61kopenevolve. Open-source implementation of AlphaEvolve
★ 6.7kCT-CLIP. Developing Generalist Foundation Models from a Multimodal Dataset for 3D Computed Tomography
★ 406Parallel-R1. The offical repo for "Parallel-R1: Towards Parallel Thinking via Reinforcement Learning"
★ 260Qwen-Agent. Agent framework and applications built upon Qwen>=3.0, featuring Function Calling, MCP, Code Interpreter, RAG, Chrome extension, etc.
★ 17kMedEvalKit. MedEvalKit: A Unified Medical Evaluation Framework
★ 246Merlin. [Nature 2026] Merlin is a 3D VLM for computed tomography that leverages both structured electronic health records (EHR) and unstructured radiology reports for pretraining.
★ 445medgemma. Jupyter Notebook
★ 1.5kgpt-oss. gpt-oss-120b and gpt-oss-20b are two open-weight language models by OpenAI
★ 20kharmony. Renderer for the harmony response format to be used with gpt-oss
★ 4.5kgemini-cli. An open-source AI agent that brings the power of Gemini directly into your terminal.
★ 106kPOLARIS. Scaling RL on advanced reasoning models
★ 689SkyRL. SkyRL: A Modular Full-stack RL Library for LLMs
★ 2.1kRL2. Python
★ 1.3kHermes-Function-Calling. Jupyter Notebook
★ 1.4kR1-Searcher-plus. R1-Searcher++: Incentivizing the Dynamic Knowledge Acquisition of LLMs via Reinforcement Learning
★ 81R1-Searcher. R1-searcher: Incentivizing the Search Capability in LLMs via Reinforcement Learning
★ 720RLLoggingBoard. A visuailzation tool to make deep understaning and easier debugging for RLHF training.
★ 294DeepEyes. Python
★ 1.2kDeepResearcher. Scaling Deep Research via Reinforcement Learning in Real-world Environments.
★ 773ReLIFT. Official Repository of "Learning what reinforcement learning can't"
★ 84Multiverse.
★ 88open-deep-research. An open source deep research clone. AI Agent that reasons large amounts of web data extracted with Firecrawl
★ 6.3knano-vllm. Nano vLLM
★ 14kAFLDM. [CVPR 2025 Oral] Alias-free Latent Diffusion Models (official implementation)
★ 107Enigmata. Resources for the Enigmata Project.
★ 82ParScale. Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling
★ 479Entropy-Mechanism-of-RL. The Entropy Mechanism of Reinforcement Learning for Large Language Model Reasoning.
★ 443MMR1. [CVPR 2026] MMR1: Enhancing Multimodal Reasoning with Variance-Aware Sampling and Open Resources
★ 217Bagel. Open-source unified multimodal model
★ 6.1ks1. s1: Simple test-time scaling
★ 6.7kinference_scaling. Python
★ 80Mixture-of-Transformers. Mixture-of-Transformers: A Sparse and Scalable Architecture for Multi-Modal Foundation Models. TMLR 2025.
★ 258Downstream-Dinov2. Downstream-Dino-V2: A GitHub repository featuring an easy-to-use implementation of the DINOv2 model by Facebook for downstream tasks such as Classification, Semantic Segmentation and Monocular depth estimation.
★ 272tips. TIPSv2 (CVPR'26) and TIPS (ICLR'25)
★ 571x-reasoner. X-Reasoner: Towards Generalizable Reasoning Across Modalities and Domains
★ 49open-thoughts. Fully open data curation for reasoning models
★ 2.3kSpeech. A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
★ 18kSEED-X. Multimodal Models in Real World
★ 557RadFM. The official code for "Towards Generalist Foundation Model for Radiology by Leveraging Web-scale 2D&3D Medical Data".
★ 559LISA. Project Page for "LISA: Reasoning Segmentation via Large Language Model"
★ 2.7kopen-instruct. AllenAI's post-training codebase
★ 3.8kSana. SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
★ 8.5kStep1X-Edit. A SOTA open-source image editing model, which aims to provide comparable performance against the closed-source models like GPT-4o and Gemini 2 Flash.
★ 2.2kminiF2F. Formal to Formal Mathematics Benchmark
★ 439tree-of-thoughts. Plug in and Play Implementation of Tree of Thoughts: Deliberate Problem Solving with Large Language Models that Elevates Model Reasoning by atleast 70%
★ 4.6ktree-of-thought-llm. [NeurIPS 2023] Tree of Thoughts: Deliberate Problem Solving with Large Language Models
★ 6kTTRL. [NeurIPS 2025] TTRL: Test-Time Reinforcement Learning
★ 1.1kml-space-benchmark. Code and data for "Does Spatial Cognition Emerge in Frontier Models?"
★ 29describe-anything. [ICCV 2025] Implementation for Describe Anything: Detailed Localized Image and Video Captioning
★ 1.5kAPR. [COLM 2025] Code for Paper: Learning Adaptive Parallel Reasoning with Language Models
★ 144litellm. Python SDK, Proxy Server (AI Gateway) to call 100+ LLM APIs in OpenAI (or native) format, with cost tracking, guardrails, loadbalancing and logging. [Bedrock, Azure, OpenAI, VertexAI, Cohere, Anthropic, Sagemaker, HuggingFace, VLLM, NVIDIA NIM]
★ 53kopencompass. OpenCompass is an LLM evaluation platform, supporting a wide range of models (Llama3, Mistral, InternLM2,GPT-4,LLaMa2, Qwen,GLM, Claude, etc) over 100+ datasets.
★ 7.2kpython-sdk. The official Python SDK for Model Context Protocol servers and clients
★ 24kThinkLite-VL. Python
★ 105Awesome-RL-based-Reasoning-MLLMs. This repository provides valuable reference for researchers in the field of multimodality, please start your exploratory travel in RL-based Reasoning MLLMs!
★ 1.4kEasyControl. Implementation of "EasyControl: Adding Efficient and Flexible Control for Diffusion Transformer"(ICCV2025)
★ 1.7kSeed-Thinking-v1.5.
★ 810PPE. Jupyter Notebook
★ 65multimodal_rewardbench. Multimodal RewardBench
★ 68Liger-Kernel. Efficient Triton Kernels for LLM Training
★ 6.5kMM-EUREKA. MM-EUREKA: Exploring the Frontiers of Multimodal Reasoning with Rule-based Reinforcement Learning
★ 771sokoban-solver. 推箱子解决方案 sokoban AI solver
★ 92VAGEN. World model reasoning RL for multi-turn VLM agents
★ 485Search-R1. Search-R1: An Efficient, Scalable RL Training Framework for Reasoning & Search Engine Calling interleaved LLM based on veRL
★ 5.1kDAPO. An Open-source RL System from ByteDance Seed and Tsinghua AIR
★ 1.8kVideo-R1. Video-R1: Reinforcing Video Reasoning in MLLMs [🔥the first paper to explore R1 for video]
★ 879radgraph. Python
★ 78GREEN. [EMNLP, Findings 2024] a radiology report generation metric that leverages the natural language understanding of language models to identify and explain clinically significant errors in candidate reports
★ 83cosmos-reason1. Cosmos-Reason1 models understand the physical common sense and generate appropriate embodied decisions in natural language through long chain-of-thought reasoning processes.
★ 950CLIP_benchmark. CLIP-like model evaluation
★ 814Self-rewarding-reasoning-LLM. Recipes to train the self-rewarding reasoning LLMs.
★ 231understand-r1-zero. Understanding R1-Zero-Like Training: A Critical Perspective
★ 1.3ksweet_rl. Benchmark and research code for the paper SWEET-RL Training Multi-Turn LLM Agents onCollaborative Reasoning Tasks
★ 271open_clip. An open source implementation of CLIP (With TULIP Support)
★ 165Prism. A Framework for Decoupling and Assessing the Capabilities of VLMs
★ 44Dynasor. [NeurIPS 2025] Simple extension on vLLM to help you speed up reasoning model without training.
★ 231atom. [NeurIPS 2025] Atom of Thoughts for Markov LLM Test-Time Scaling
★ 663Open-Reasoner-Zero. Official Repo for Open-Reasoner-Zero
★ 2.1kVisualSketchpad. Codes for Visual Sketchpad: Sketching as a Visual Chain of Thought for Multimodal Language Models
★ 287PixelRefer. The code for PixelRefer & VideoRefer
★ 352RAGEN. RAGEN leverages reinforcement learning to train LLM reasoning agents in interactive, stochastic environments.
★ 2.7kVideoGLaMM. [CVPR 2025 🔥]A Large Multimodal Model for Pixel-Level Visual Grounding in Videos
★ 104groundingLMM. [CVPR 2024 🔥] Grounding Large Multimodal Model (GLaMM), the first-of-its-kind model capable of generating natural language responses that are seamlessly integrated with object segmentation masks.
★ 963camel. 🐫 CAMEL: The first and the best multi-agent framework. Finding the Scaling Law of Agents. https://www.camel-ai.org
★ 17kowl. 🦉 OWL: Optimized Workforce Learning for General Multi-Agent Assistance in Real-World Task Automation
★ 20kVisualThinker-R1-Zero. Explore the Multimodal “Aha Moment” on 2B Model
★ 624Artemis. [NeurIPS 2024] Artemis: Towards Referential Understanding in Complex Videos
★ 27densecap. Dense image captioning in Torch
★ 1.6kPOPE. The official GitHub page for ''Evaluating Object Hallucination in Large Vision-Language Models''
★ 265ALIA. Augmenting with Language-guided Image Augmentation (ALIA)
★ 78open-infra-index. Production-tested AI infrastructure tools for efficient AGI development and community-driven innovation
★ 8kEasyOCR. Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
★ 30kX-R1. minimal-cost for training 0.5B R1-Zero
★ 815Awesome-ML-SYS-Tutorial. My learning notes for ML SYS.
★ 6.7klmms-eval. One-for-All Multimodal Evaluation Toolkit Across Text, Image, Video, and Audio Tasks
★ 4.3kLIMR. Python
★ 220MoBA. MoBA: Mixture of Block Attention for Long-Context LLMs
★ 2.1kflashinfer. FlashInfer: Kernel Library for LLM Serving
★ 5.9kLLaVA-MORE. [ICCVW 25] LLaVA-MORE: A Comparative Study of LLMs and Visual Backbones for Enhanced Visual Instruction Tuning
★ 160ubuntu-slurm. Steps to create a small slurm cluster with GPU enabled nodes
★ 272rllm. Democratizing Reinforcement Learning for LLMs
★ 5.7kReasonFlux. [NeurIPS 2025 Spotlight] LLM post-training suite — featuring ReasonFlux, ReasonFlux-PRM, and ReasonFlux-Coder.
★ 539T1. RL Scaling and Test-Time Scaling (ICML'25)
★ 115open-r1. Fully open reproduction of DeepSeek-R1
★ 26kdemystify-long-cot. Python
★ 336MMVP. Python
★ 363open-r1-multimodal. A fork to add multimodal model training to open-r1
★ 1.6kR1-V. Witness the aha moment of VLM with less than $3.
★ 4.1kpeft. 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
★ 21kSoM. [arXiv 2023] Set-of-Mark Prompting for GPT-4V and LMMs
★ 1.5ksimpleRL-reason. Simple RL training for reasoning
★ 3.9kMAVIS. [ICLR 2025] Mathematical Visual Instruction Tuning for Multi-modal Large Language Models
★ 156gRefCOCO. A benchmark dataset for GREx: GRES, GREC, and GREG [CVPR 2023 & IJCV 2026]
★ 241Ref-L4. Evaluation code for Ref-L4, a new REC benchmark in the LMM era
★ 61TinyZero. Minimal reproduction of DeepSeek R1-Zero
★ 13kml-ferret. Python
★ 8.7kanole. [Extended verision ICLR 2025 Blog Track] Anole: An Open, Autoregressive and Native Multimodal Models for Interleaved Image-Text Generation
★ 841MiniMax-01. The official repo of MiniMax-Text-01 and MiniMax-VL-01, large-language-model & vision-language-model based on Linear Attention
★ 3.4kKimi-k1.5.
★ 3.5kDeepSeek-R1.
★ 92kLlamaFactory. Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
★ 73kvllm. A high-throughput and memory-efficient inference and serving engine for LLMs
★ 86kM3CoT. Python
★ 92thinking-in-space. Official repo and evaluation implementation of VSI-Bench
★ 731coconut. Training Large Language Model to Reason in a Continuous Latent Space
★ 1.7kVisual-CoT. [Neurips'24 Spotlight] Visual CoT: Advancing Multi-Modal Language Models with a Comprehensive Dataset and Benchmark for Chain-of-Thought Reasoning
★ 447instruct-pix2pix. Python
★ 6.9kms-swift. Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.6, DeepSeek-V4, GLM-5.1, InternLM3, Llama4, ...) and 300+ MLLMs (Qwen3-VL, Qwen3-Omni, InternVL3.5, Ovis2.5, GLM4.5v, Gemma4, Llava, Phi4, ...) (AAAI 2025).
★ 15kPRIME. Scalable RL solution for advanced reasoning of language models
★ 1.9kverl. verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework
★ 22kSkyThought. Sky-T1: Train your own O1 preview model within $450
★ 3.4kFasterViT. [ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
★ 921Slow_Thinking_with_LLMs. A series of technical report on Slow Thinking with LLM
★ 767Virgo. Official code of *Virgo: A Preliminary Exploration on Reproducing o1-like MLLM*
★ 110VLMEvalKit. Open-source evaluation toolkit of large multi-modality models (LMMs), support 220+ LMMs, 80+ benchmarks
★ 4.3kIG-LLM. Code for "Re-Thinking Inverse Graphics With Large Language Models"; TMLR 2024
★ 71SWE-Gym. Code for Paper: Training Software Engineering Agents and Verifiers with SWE-Gym [ICML 2025]
★ 701Mask2Former. Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"
★ 3.4kprismer. The implementation of "Prismer: A Vision-Language Model with Multi-Task Experts".
★ 1.3kCrossFlow. [CVPR2025] PyTorch-based reimplementation of CrossFlow, as proposed in 'Flowing from Words to Pixels: A Noise-Free Framework for Cross-Modality Evolution'
★ 342taskonomy. Taskonomy: Disentangling Task Transfer Learning [Best Paper, CVPR2018]
★ 877BLINK_Benchmark. This repo contains evaluation code for the paper "BLINK: Multimodal Large Language Models Can See but Not Perceive". https://arxiv.org/abs/2404.12390 [ECCV 2024]
★ 170RADIO. Official repository for "AM-RADIO: Reduce All Domains Into One"
★ 1.9kLibreChat. Enhanced ChatGPT Clone: Features Agents, MCP, Skills, DeepSeek, Anthropic, AWS, OpenAI, Responses API, Azure, Groq, o1, GPT-5, Mistral, OpenRouter, Vertex AI, Gemini, Artifacts, AI model switching, message search, Code Interpreter, langchain, DALL-E-3, OpenAPI Actions, Functions, Secure Multi-User Auth, Presets, open-source for self-hosting. Active
★ 41kstream-of-search. Repository for the paper Stream of Search: Learning to Search in Language
★ 154FreeNoise. [ICLR 2024] Code for FreeNoise based on VideoCrafter
★ 429RAM. A framework to study AI models in Reasoning, Alignment, and use of Memory (RAM).
★ 377lingua. Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.
★ 4.8kLVOS.
★ 92MOSE-api. [ICCV 2023] MOSE: A New Dataset for Video Object Segmentation in Complex Scenes
★ 384kinetics-dataset. Shell
★ 977sd-scripts. Python
★ 7.2kMegatron-LM. Ongoing research training transformer models at scale
★ 17kx-flux. Python
★ 2.2kfluxgym. Dead simple FLUX LoRA training UI with LOW VRAM support
★ 3.2ksam-home. Segment Anything (SAM) at Home web app using Gradio
★ 14huggingface_hub. The official Python client for the Hugging Face Hub.
★ 3.7ksegment-anything-demo-react-fastapi. This repository is a sample implementation of frontend/backend using SAM code from meta.
★ 34meta-sam-demo. Meta's Segment Anything Model (SAM) Demo Site
★ 153PyramidDrop. (CVPR 2025) PyramidDrop: Accelerating Your Large Vision-Language Models via Pyramid Visual Redundancy Reduction
★ 151nerfstudio. A collaboration friendly studio for NeRFs
★ 12knerf. Code release for NeRF (Neural Radiance Fields)
★ 11kOpenVid-1M. [ICLR 2025] OpenVid-1M: A Large-Scale High-Quality Dataset for Text-to-video Generation
★ 450Draw-and-Understand. [ICLR2025] Draw-and-Understand: Leveraging Visual Prompts to Enable MLLMs to Comprehend What You Want
★ 94Osprey. [CVPR2024] The code for "Osprey: Pixel Understanding with Visual Instruction Tuning"
★ 842