IBM / multiple-input-multiple-output-netsLinks
PyTorch Implementation of the paper "MIMONets: Multiple-Input-Multiple-Output Neural Networks Exploiting Computation in Superposition" published in NeurIPS 2023.
☆26Updated 4 months ago
Alternatives and similar repositories for multiple-input-multiple-output-nets
Users that are interested in multiple-input-multiple-output-nets are comparing it to the libraries listed below
Sorting:
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆82Updated last year
- ☆69Updated 10 months ago
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- ☆129Updated 6 months ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" (ICLR 2025)☆67Updated 10 months ago
- Software for using predictive coding algorithms to train PyTorch models.☆26Updated last year
- ☆62Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated last year
- Spyx: Spiking Neural Networks in JAX☆131Updated last month
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆35Updated 4 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated 2 years ago
- This is the code repository associated with the paper "Abstractors and relational cross-attention: An inductive bias for explicit relati…☆21Updated last year
- ☆12Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆46Updated 2 years ago
- ☆27Updated 3 years ago
- This repository contains code for the paper "Learning Decision Trees as Amortized Structure Inference"☆15Updated 10 months ago
- Spiking neuron integration for PyTorch☆41Updated 10 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated last year
- Uncertainty-aware representation learning (URL) benchmark☆106Updated 10 months ago
- ☆20Updated 2 years ago
- ☆33Updated last year
- Oscillatory State-Space Models☆113Updated 3 months ago
- ☆28Updated 2 years ago
- Code accompanying the paper "A contrastive rule for meta-learning"☆13Updated last year
- State Space Models☆72Updated last year
- We integrate discrete diffusion models with neurosymbolic predictors for scalable and calibrated learning and reasoning☆59Updated 2 months ago