IDSIA / neuraldiffeq-fwp
Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (NeurIPS 2022)
☆20Updated 2 years ago
Alternatives and similar repositories for neuraldiffeq-fwp:
Users that are interested in neuraldiffeq-fwp are comparing it to the libraries listed below
- The Energy Transformer block, in JAX☆56Updated last year
- ☆34Updated 3 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- flexible meta-learning in jax☆12Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Updated last year
- Contact-Aware Symplectic Integrator Network☆12Updated last year
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆48Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 3 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- ☆30Updated 5 months ago
- ☆18Updated 3 years ago
- ☆37Updated 2 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆70Updated last year
- Bayesian model reduction for probabilistic machine learning☆10Updated 2 weeks ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆26Updated 10 months ago
- ☆49Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- symbolic regression☆39Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- ☆15Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Variational Reinforcement Learning☆16Updated 7 months ago
- Neuroevolution Benchmark in JAX 🦕☆38Updated last year
- A Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆14Updated last month