tristandeleu / jax-comln
Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)
☆27Updated 2 years ago
Alternatives and similar repositories for jax-comln:
Users that are interested in jax-comln are comparing it to the libraries listed below
- Simple, extensible implementations of some meta-learning algorithms in Jax☆9Updated 4 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 5 months ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆18Updated 2 years ago
- Generalised UDRL☆37Updated 2 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆38Updated 4 years ago
- Probabilistic Solution of Differential Equations☆13Updated 2 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Contact-Aware Symplectic Integrator Network☆11Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 3 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆35Updated 3 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- ☆10Updated 2 years ago
- Public Release of Plan2vec Implementation in pyTorch☆56Updated 2 years ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- Open source code for paper "On the Learning and Learnability of Quasimetrics".☆32Updated 2 years ago
- This is the official source code for Sequential Neural Processes.☆40Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- ☆34Updated 3 years ago
- PyTorch Package For Quasimetric Learning☆41Updated 2 months ago
- ☆17Updated 3 years ago
- ☆55Updated 2 years ago
- A pytorch implementation of Amortized Stein Variational Gradient Descent/ Stein GAN☆18Updated 6 years ago
- Code for "MetaFun: Meta-Learning with Iterative Functional Updates"☆14Updated 4 years ago