Skip to content
View gyu-heo's full-sized avatar
  • Sabatini Lab, Harvard Medical School
  • Boston

Highlights

  • Pro

Block or report gyu-heo

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
Showing results

Welcome to the Physics-based Deep Learning Book (v0.2)

Jupyter Notebook 1,051 165 Updated Dec 9, 2024

A dataloader, but for JAX

Python 20 Updated May 17, 2024

JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️

Python 266 19 Updated Nov 16, 2024

Accelerated minigrid environments with JAX

Python 132 16 Updated Aug 1, 2024
Python 147 16 Updated Mar 5, 2025

Implementation of Dreamer v3 in pytorch.

Python 505 112 Updated Sep 27, 2024

Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselines.

Python 47 5 Updated Feb 9, 2025

SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms

Python 405 37 Updated Mar 8, 2025

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

Jupyter Notebook 10,661 1,385 Updated Nov 4, 2024

⚡ Flashbax: Accelerated Replay Buffers in JAX

Python 227 15 Updated Feb 25, 2025

An implementation of DreamerV2 written in JAX and Haiku.

Python 1 1 Updated Dec 14, 2022

An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.

Python 15 1 Updated Jan 16, 2023

Learning Robust Dynamics Through Variational Sparse Gating

Python 21 1 Updated Oct 19, 2022

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

Python 2,586 341 Updated Mar 9, 2025

PyTorch based implementation of model-based RL algorithm PlaNet

Python 3 Updated Sep 1, 2023

LLMs in JAX

Python 6 Updated Mar 27, 2024

Evaluating long-term memory of reinforcement learning algorithms

Python 140 16 Updated Jun 23, 2023

PyTorch implementation of DreamerV2 model-based RL algorithm

Python 216 49 Updated Apr 26, 2023

Deep Planning Network: Control from pixels by latent planning with learned dynamics

Python 370 59 Updated Oct 15, 2021

Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper

Python 15 4 Updated Jul 14, 2024
Python 2 Updated Aug 31, 2022

PyTorch implementation of Asynchronous Advantage Actor Critic (A3C) from "Asynchronous Methods for Deep Reinforcement Learning".

Python 1,251 279 Updated Sep 25, 2019

PyTorch implementation of Episodic Meta Reinforcement Learning on variants of the "Two-Step" task. Reproduces the results found in three papers. Check the ReadMe for more details!

Python 34 3 Updated Dec 12, 2020

PlaNet: Learning Latent Dynamics for Planning from Pixels

Python 10 1 Updated Feb 13, 2020

Pytorch implementation of Dreamer-v2: Visual Model Based RL Algorithm.

Python 253 43 Updated Jul 29, 2023

Learning Latent Dynamics for Planning from Pixels

Python 1,186 203 Updated Mar 24, 2023

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

Python 700 89 Updated Feb 4, 2025

Simple and easily configurable grid world environments for reinforcement learning

Python 2,188 618 Updated Feb 6, 2025

MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research

Python 489 60 Updated Feb 12, 2025

Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"

Jupyter Notebook 13 3 Updated Jun 16, 2024
Next