dylandoblar / noether-networks
Meta-learning inductive biases in the form of useful conserved quantities.
☆37Updated 2 years ago
Alternatives and similar repositories for noether-networks:
Users that are interested in noether-networks are comparing it to the libraries listed below
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- ☆24Updated 6 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- A JAX implementation of stochastic addition.☆14Updated 2 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆24Updated 4 years ago
- ☆24Updated 2 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Updated last year
- A port of muP to JAX/Haiku☆25Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated last year
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆27Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆20Updated 2 years ago
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Official code for the paper "Compositional Generalization from First Principles" (NeurIPS 2023)☆10Updated last year
- PyTorch Package For Quasimetric Learning☆41Updated 4 months ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆68Updated this week
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- ☆26Updated 2 years ago
- ☆34Updated 3 years ago
- ☆49Updated 4 years ago
- The official repository for our paper "The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformers". We s…☆67Updated 2 years ago
- Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)☆48Updated 2 years ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆19Updated last year