jannerm Goto Github PK

bairblog.github.io

bullet-manipulation

Robotic manipulation in PyBullet

bullet3

Bullet Physics SDK: real-time collision detection and multi-physics simulation for VR, games, visual effects, robotics, machine learning etc.

d4rl

A benchmark for offline reinforcement learning.

ddpo

Code for the paper "Training Diffusion Models with Reinforcement Learning"

diffuser

Code for the paper "Planning with Diffusion for Flexible Behavior Synthesis"

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

doodad

face-intrinsics

gamma-blog

gamma-models

Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"

gumbel_softmax_vae

PyTorch implementation of a Variational Autoencoder with Gumbel-Softmax Distribution

gym_minigrid

intrinsics-network

Code for the paper "Self-Supervised Intrinsic Image Decomposition"

mbpo

Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"

mbpo-www

mujoco-py

MuJoCo is a physics engine for detailed, efficient rigid body simulations with contacts. mujoco-py allows using MuJoCo from Python 3.

o2p2

ray

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a toolkit of libraries (Ray AIR) for accelerating ML workloads.

rlkit

Collection of reinforcement learning algorithms

simplesac

A simple and easy to use implementation of the soft actor-critic algorithm.

slice-sampling

spatial-reasoning

Code for the paper "Representation Learning for Grounded Spatial Reasoning"

torch-colormap

MATLAB-style colormaps for Torch

trajectory-transformer

Code for the paper "Offline Reinforcement Learning as One Big Sequence Modeling Problem"

viskit

rllab's viskit with some added features