akandykeller / NeuralWaveMachines
Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally Coupled Oscillatory Recurrent Neural Networks"
☆69Updated last year
Alternatives and similar repositories for NeuralWaveMachines:
Users that are interested in NeuralWaveMachines are comparing it to the libraries listed below
- Quantify geometric intelligence in natural and artificial brains.☆50Updated 2 months ago
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆42Updated 8 months ago
- ☆33Updated last year
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025).☆81Updated 3 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 10 months ago
- ☆37Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- The Energy Transformer block, in JAX☆56Updated last year
- Hierarchical Associative Memory User Experience☆98Updated last year
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆35Updated 3 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆27Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆97Updated 2 years ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆36Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆19Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Flow-matching algorithms in JAX☆83Updated 6 months ago
- ☆40Updated 2 months ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Running Jax in PyTorch Lightning☆86Updated 2 months ago
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆43Updated 3 years ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆164Updated last year
- Code for minimum-entropy coupling.☆31Updated 7 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- ☆59Updated 2 years ago
- ☆48Updated 2 years ago
- Diffusion models in PyTorch☆92Updated this week