puyuan1996

PromptAgent

This is the official repo for "PromptAgent: Strategic Planning with Language Models Enables Expert-level Prompt Optimization". PromptAgent is a novel automatic prompt optimization method that autonomously crafts prompts equivalent in quality to those handcrafted by experts, i.e., expert-level prompts.

FoRL

A library for for First-order Reinforcement Learning algorithms

PWM

PWM: Policy Learning with Large World Models

nano-llama31

nanoGPT style version of Llama 3.1

graphrag

A modular graph-based Retrieval-Augmented Generation (RAG) system

DeepSeek-V2

DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Model

onnx

Open standard for machine learning interoperability

XTRA

On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning

ellm

lm-human-preferences

Code for the paper Fine-Tuning Language Models from Human Preferences

trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

RL4LMs

A modular RL library to fine-tune language models to human preferences

OpenRLHF

An Easy-to-use, Scalable and High-performance RLHF Framework (70B+ PPO Full Tuning & Iterative DPO & LoRA & Mixtral)

dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

mistral-inference

Official inference library for Mistral models

soft-moe-pytorch

Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch

torchtune

A Native-PyTorch Library for LLM Fine-tuning

LanguageAgentTreeSearch

[ICML 2024] Official repository for "Language Agent Tree Search Unifies Reasoning Acting and Planning in Language Models"

MCTS-DPO

This is the repository that contains the source code for the Self-Evaluation Guided MCTS for online DPO.

audiocraft

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable music generation LM with textual and melodic conditioning.

rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

rl-learned-optimization

Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"

godot

Godot Engine – Multi-platform 2D and 3D game engine

avalon

A 3D video game environment and benchmark designed from scratch for reinforcement learning research

GenerativeRL

Python library for solving reinforcement learning (RL) problems using generative models (e.g. Diffusion Models).

introRL

Intro to Reinforcement Learning (强化学习纲要）

Gymnasium

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

glow

Code for reproducing results in "Glow: Generative Flow with Invertible 1x1 Convolutions"

SiT

Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"

gemma_pytorch

The official PyTorch implementation of Google's Gemma models