tristandeleu / jax-comln
Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)
☆26Updated 3 years ago
Alternatives and similar repositories for jax-comln:
Users that are interested in jax-comln are comparing it to the libraries listed below
- Variational Reinforcement Learning☆16Updated 9 months ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years 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
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Generalised UDRL☆37Updated 2 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…☆21Updated 4 years ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 4 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- Code for "MetaFun: Meta-Learning with Iterative Functional Updates"☆14Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- flexible meta-learning in jax☆13Updated last year
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 4 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- Contact-Aware Symplectic Integrator Network☆14Updated 2 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Updated 6 years ago
- Code for Deep Reinforcement and InfoMax Learning (Neurips 2020)☆10Updated 4 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- ☆53Updated 9 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆24Updated 6 months ago
- LaTeX source code for the slides☆23Updated 3 years ago
- Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen☆25Updated 4 years ago
- ☆19Updated 3 years ago