ZhengyaoJiang / GTG
Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
☆27Updated 3 years ago
Related projects: ⓘ
- Variational Reinforcement Learning☆16Updated last month
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated last year
- Generalised UDRL☆37Updated 2 years ago
- Code to reproduce results on toy tasks and companion blog for the paper.☆20Updated 2 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆14Updated last year
- ☆29Updated 2 years ago
- JAX implementation of Graph Attention Networks☆12Updated 2 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated last year
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated last year
- [ICLR 2020, Oral] Harnessing Structures for Value-Based Planning and Reinforcement Learning☆34Updated 4 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Updated 5 years ago
- Code for paper "Model-based Adversarial Meta-Reinforcement Learning" (https://arxiv.org/abs/2006.08875)☆32Updated 3 years ago
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆39Updated 2 weeks ago
- Code for Dataset and Benchmarks Submission, Neurips 2022☆13Updated 2 years ago
- A Continual Multi-agent RL testbed based on Hanabi☆30Updated 3 years ago
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- Contextual Bandits Action Elimination DQN☆19Updated 6 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆24Updated 2 years ago
- ☆49Updated 3 years ago
- ☆18Updated 3 years ago
- TaskMet Task-driven Metric Learning for Model Learning☆18Updated 7 months ago
- Code for Quantifying Ignorance in Individual-Level Causal-Effect Estimates under Hidden Confounding☆21Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆47Updated last year
- Code for our paper: Hierarchical RL Using an Ensemble of Proprioceptive Periodic Policies☆15Updated 5 years ago
- A PyTorch implementation of "Meta-Amortized Variational Inference and Learning" (https://arxiv.org/abs/1902.01950)☆14Updated 4 years ago
- Taming MAML: efficient unbiased meta-reinforcement learning☆28Updated last year
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆20Updated 3 years ago
- ☆17Updated 2 years ago