ZhengyaoJiang / GTG
Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).
☆27Updated 3 years ago
Alternatives and similar repositories for GTG:
Users that are interested in GTG are comparing it to the libraries listed below
- Variational Reinforcement Learning☆16Updated 8 months ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- Generalised UDRL☆37Updated 2 years ago
- ☆29Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 6 years ago
- Source code for paper Conservative Uncertainty Estimation By Fitting Prior Networks (ICLR 2020)☆21Updated 2 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 4 years ago
- An adaptive training algorithm for residual network☆15Updated 4 years ago
- Code for "Learning Inductive Biases with Simple Neural Networks" (Feinman & Lake, 2018).☆21Updated 6 years ago
- Energy Based Models are a quite novel technique for density estimation. In this university project I explore this new research topic and …☆16Updated 3 years ago
- Library for Auto-Encoding Sequential Monte Carlo☆18Updated last year
- ☆19Updated 3 years ago
- ☆49Updated 4 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- Code Release for Task Agnostic Dynamics Priors for Deep Reinforcement Learning☆12Updated 5 years ago
- This is an article about using variational autoencoders for the generation of new data. It contains the code for generating the plots and…☆12Updated 4 years ago
- Contact-Aware Symplectic Integrator Network☆12Updated 2 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- ☆15Updated 4 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Updated 6 years ago
- Implementation of Sum-Product Attend-Infer-Repeat☆30Updated 4 years ago
- ☆19Updated 3 years ago
- [ICLR 2020, Oral] Harnessing Structures for Value-Based Planning and Reinforcement Learning☆33Updated 5 years ago
- Code for Deep Reinforcement and InfoMax Learning (Neurips 2020)☆10Updated 4 years ago