tristandeleu / jax-comlnLinks
Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)
☆25Updated 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
Sorting:
- Variational Reinforcement Learning☆16Updated last year
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆20Updated 5 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 5 years ago
- Public Release of Plan2vec Implementation in pyTorch☆57Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Updated 5 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Probabilistic Solution of Differential Equations☆13Updated 3 years ago
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆57Updated 4 years ago
- ☆35Updated 3 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 4 years ago
- The Differentiable Cross-Entropy Method☆124Updated 5 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- Tutorial on amortized optimization for learning to optimize over continuous domains☆246Updated 3 weeks ago
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- Generalised UDRL☆37Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- Convex potential flows☆84Updated 3 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 4 years ago
- Decentralized Reinforcment Learning: Global Decision-Making via Local Economic Transactions (ICML 2020)☆43Updated 2 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- Plannable Approximations to MDP Homomorphisms: Equivariance under Actions☆30Updated 5 years ago
- Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"☆45Updated 2 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆19Updated 6 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Updated 2 years ago
- ☆43Updated 2 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆56Updated 4 years ago