akandykeller / NeuralWaveMachines
Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally Coupled Oscillatory Recurrent Neural Networks"
☆70Updated 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 4 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated last month
- ☆33Updated 2 years ago
- ☆60Updated 3 years ago
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆43Updated 10 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- ☆10Updated last month
- The Energy Transformer block, in JAX☆56Updated last year
- Software for using predictive coding algorithms to train PyTorch models.☆21Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- ☆42Updated 3 weeks ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆27Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆100Updated 2 years ago
- ☆38Updated 2 years ago
- Flow-matching algorithms in JAX☆87Updated 8 months ago
- Graph neural networks in JAX.☆67Updated 9 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025)☆84Updated last week
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Hierarchical Associative Memory User Experience☆101Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆32Updated last week
- 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 the paper: Rotating Features for Object Discovery☆50Updated 8 months ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆43Updated 5 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated 2 years ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- ☆192Updated 10 months ago