Jong-Kook Heo's repositories
causal-confusion
A toolkit for developing and comparing reinforcement learning algorithms.
control-pcgrl
Train or evolve controllable and diverse level-generators.
CQL
PyTorch implementation of the Offline Reinforcement Learning algorithm CQL. Includes the versions DQN-CQL and SAC-CQL for discrete and continuous action spaces.
DeepRL_PyTorch
Deep Reinforcement Learning codes for study. Currently, there are only codes for algorithms: DQN, C51, QR-DQN, IQN, QUOTA.
DPPO
Official implementation of "Direct Preference-based Policy Optimization without Reward Modeling" (NeurIPS 2023)
epic
Implements the Equivalent-Policy Invariant Comparison (EPIC) distance for reward functions.
gym-pcgrl
A package for "Procedural Content Generation via Reinforcement Learning" OpenAI Gym interface.
Kernel-based-Learning
Kernel SVM Implementation for Molecule Dataset
mario-gpt
[Neurips 2023] Generating Mario Levels with GPT2. Code for the paper "MarioGPT: Open-Ended Text2Level Generation through Large Language Models" https://arxiv.org/abs/2302.05981
ms-level-gen
Start Small: Training Controllable Game Level Generators without Training Data by Learning at Multiple Sizes
multi-sensor-FDII-health-forecasting-for-autonomous-vehicles
An Extension of A2D2 dataset by Audi with augmented sensor faults and different degradation paths
oprl
Official Codebase for TMLR 2023, Benchmarks and Algorithms for Offline Preference-Based Reward Learning
paired
PAIRED in PyTorch 🔥
PALM-E
Implementation of "PaLM-E: An Embodied Multimodal Language Model"
PaLM-rlhf-pytorch
Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
robotic-transformer-pytorch
Implementation of RT1 (Robotic Transformer) in Pytorch
rune
Code for paper: Reward Uncertainty for Exploration in Preference-based Reinforcement Learning
Semi-SupervisedLearning
SemiSupervisedLearning
TorchSSL
A PyTorch-based library for semi-supervised learning (NeurIPS'21)