hadivafaii / IterativeVAELinks
Brain-like variational inference
☆51Updated last month
Alternatives and similar repositories for IterativeVAE
Users that are interested in IterativeVAE are comparing it to the libraries listed below
Sorting:
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆47Updated 2 weeks ago
- We integrate discrete diffusion models with neurosymbolic predictors for scalable and calibrated learning and reasoning☆37Updated last month
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆38Updated last year
- Software for using predictive coding algorithms to train PyTorch models.☆25Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- ☆39Updated 3 years ago
- ☆12Updated last week
- Scalable and Stable Parallelization of Nonlinear RNNS☆16Updated 4 months ago
- Quantify geometric intelligence in natural and artificial brains.☆51Updated 6 months ago
- ☆32Updated last year
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- Pytorch optimiser for training ANNs with exponentiated gradient desent☆18Updated 3 months ago
- Reconstructing shared causal drivers from noisy time series☆54Updated 8 months ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated last year
- The Zebrafish Activity Prediction Benchmark measures progress on the problem of predicting cellular-resolution neural activity throughout…☆41Updated 2 weeks ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- ☆32Updated 8 months ago
- ☆53Updated 8 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆30Updated last year
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆44Updated 7 months ago
- Official repository of "Spontaneous symmetry breaking in generative diffusion models"☆43Updated last year
- ☆51Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 5 months ago
- ☆18Updated last year
- ☆38Updated last year
- The code for "Diffusion Geometry" (2024).☆43Updated last year
- Predictive Coding JAX-based library☆69Updated 2 months ago
- ☆104Updated 2 weeks ago