TrentBrick / attention-approximates-sdm
Code for the paper "Attention Approximates Sparse Distributed Memory"
☆22Updated 3 years ago
Alternatives and similar repositories for attention-approximates-sdm:
Users that are interested in attention-approximates-sdm are comparing it to the libraries listed below
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆69Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Software for using predictive coding algorithms to train PyTorch models.☆18Updated 11 months ago
- The Energy Transformer block, in JAX☆54Updated last year
- ☆14Updated 3 years ago
- ☆24Updated last year
- ☆17Updated 3 years ago
- Fully documented Pytorch implementation of the Equilibrium Propagation algorithm.☆31Updated 5 years ago
- Implementation of the unbounded depth neural network from the paper Variational Inference for Infinitely Deep Neural Networks☆17Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆59Updated 3 years ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- Quantify geometric intelligence in natural and artificial brains.☆50Updated 2 months ago
- A lightweight and flexible framework for Hebbian learning in PyTorch.☆81Updated 11 months ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- Image augmentation library for Jax☆37Updated 9 months ago
- ☆29Updated 3 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆38Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆42Updated 4 years ago
- ☆31Updated 9 months ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆35Updated last year
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- ☆37Updated last year
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆27Updated 2 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆24Updated 2 years ago
- An implementation of Resonator Networks☆19Updated 4 years ago
- Bayesian model reduction for probabilistic machine learning☆10Updated last month