Awni00 / abstractorLinks
This is the code repository associated with the paper "Abstractors and relational cross-attention: An inductive bias for explicit relational reasoning in Transformers"
☆21Updated last year
Alternatives and similar repositories for abstractor
Users that are interested in abstractor are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆61Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆93Updated 3 years ago
- repo for code for paper on general theory associative memory models☆20Updated 3 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆84Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆43Updated 2 years ago
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Updated 2 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆57Updated 2 years ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆105Updated last month
- ☆25Updated last year
- Code for the paper: "Independent mechanism analysis, a new concept?"☆24Updated 2 years ago
- Predictive Coding JAX-based library☆84Updated 7 months ago
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆83Updated last year
- ☆27Updated 2 years ago
- Transformers with doubly stochastic attention☆50Updated 3 years ago
- ☆34Updated 7 months ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- ☆27Updated 2 years ago
- A modular, easy to extend GFlowNet library☆295Updated last week
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆45Updated 3 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 4 years ago
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆11Updated 4 months ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆66Updated 3 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆34Updated last month
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆46Updated 2 years ago
- Mutual information estimators and benchmark☆54Updated 2 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- ☆14Updated 4 years ago