gpt-rl Goto Github PK

acme

A library of reinforcement learning components and agents

babyai

BabyAI platform. A testbed for training agents to understand and execute language commands.

clevr_robot_env

CLEVR-Robot: a reinforcement learning environment combining vision, language and control.

decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

fsvf

fsvf-toy

generalization

generalization-old

gpt-neox-rest

GPT-NeoX modified to run inference via a REST API

gym-minigrid

Minimalistic gridworld package for OpenAI Gym

gym-miniworld

Simple 3D interior simulator for RL & robotics research

habitat-lab

A modular high-level library to train embodied AI agents across a variety of tasks, environments, and simulators.

habitat-sim

A flexible, high-performance 3D simulator for Embodied AI research.

hal

PyTorch Implementation of "Language as an Abstraction for Hierarchical Deep Reinforcement Learning" paper

impala

ppo

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

pybullet-urdf-models

Collection of urdf models (including part of YCB dataset). Made for robot manipulation and grasping simulation, tested with PyBullet.

pyrep

A toolkit for robot learning research.

pytorch-a2c-ppo-acktr

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO) and Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR).

r2d2

An Implementation of Recurrent Experience Replay in Distributed Reinforcement Learning (Kapturowski et al. 2019) in PyTorch and Ray

rlbench

A large-scale benchmark and learning environment.

sac-discrete.pytorch

A PyTorch implementation of SAC-Discrete.

stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

torchbeast

A PyTorch Platform for Distributed RL

transformers

🤗 Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

ycb-tools

Tools to download models from the YCB dataset and use them with the Gazebo simulator.