IDSIA / neuraldiffeq-fwpLinks
Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (NeurIPS 2022)
☆22Updated last week
Alternatives and similar repositories for neuraldiffeq-fwp
Users that are interested in neuraldiffeq-fwp are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆58Updated last year
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆28Updated last year
- ☆16Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- Variational Reinforcement Learning☆16Updated 11 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- ☆34Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- flexible meta-learning in jax☆14Updated last year
- Bayesian model reduction for probabilistic machine learning☆11Updated 2 months ago
- ☆19Updated 3 years ago
- ☆32Updated 8 months ago
- ☆39Updated 3 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆49Updated last week
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Neural Optimal Transport with Lagrangian Costs☆56Updated last month
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Updated 3 years ago
- Neuroevolution Benchmark in JAX 🦕☆39Updated last year
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆26Updated 4 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆32Updated 3 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- ☆12Updated 11 months ago
- ☆17Updated 10 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- The implementation of "The Kanerva Machine" with Pytorch and Pyro☆12Updated 7 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago