Stars
Making large AI models cheaper, faster and more accessible
Train transformer language models with reinforcement learning.
RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RN…
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
An educational resource to help anyone learn deep reinforcement learning.
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)
Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image …
Common used path planning algorithms with animations.
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
A library of reinforcement learning components and agents
An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)
OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.
A high-performance distributed training framework for Reinforcement Learning
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Implementations of basic RL algorithms with minimal lines of codes! (pytorch based)
Motion planning algorithms commonly used on autonomous vehicles. (path planning + path tracking)
PPO x Family DRL Tutorial Course（决策智能入门级公开课：8节课帮你盘清算法理论，理顺代码逻辑，玩转决策AI应用实践 ）
Author's PyTorch implementation of TD3 for OpenAI gym tasks
Elucidating the Design Space of Diffusion-Based Generative Models (EDM)
Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
PaddleSlim is an open-source library for deep model compression and architecture search.