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 5 months ago
- The Energy Transformer block, in JAX☆57Updated last year
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆43Updated 11 months ago
- ☆33Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- ☆39Updated 3 years ago
- Hierarchical Associative Memory User Experience☆101Updated last year
- ☆60Updated 3 years ago
- Quantification of Uncertainty with Adversarial Models☆28Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆26Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated 2 months ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Normalizing Flows using JAX☆83Updated last year
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆28Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆70Updated last week
- Deep Networks Grok All the Time and Here is Why☆34Updated 11 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- ☆43Updated last month
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- ☆17Updated 8 months ago
- nanoGPT using Equinox☆13Updated 2 years ago
- ☆11Updated 2 weeks ago
- ☆34Updated 3 years ago
- Portfolio REgret for Confidence SEquences☆14Updated 4 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆32Updated 3 weeks ago