bhoov / hamuxLinks
Hierarchical Associative Memory User Experience
☆106Updated this week
Alternatives and similar repositories for hamux
Users that are interested in hamux are comparing it to the libraries listed below
Sorting:
- Running Jax in PyTorch Lightning☆118Updated last year
- ☆250Updated 6 months ago
- A functional training loops library for JAX☆88Updated last year
- ☆194Updated last week
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆77Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- ☆118Updated last month
- Neural Networks for JAX☆84Updated last year
- Predictive Coding JAX-based library☆86Updated 8 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆165Updated last year
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆98Updated last month
- Flexible Inference for Predictive Coding Networks in JAX.☆65Updated this week
- Code for the paper "Predictive Coding Approximates Backprop along Arbitrary Computation Graphs"☆167Updated 5 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- ☆162Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆191Updated this week
- JAX Arrays for human consumption☆110Updated 2 months ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Updated last year
- Automatic gradient descent☆216Updated 2 years ago
- Bayesian inference with Python and Jax.☆34Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- ☆60Updated 3 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep R…☆174Updated 7 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- ☆112Updated 3 years ago
- Graph neural networks in JAX.☆68Updated last year
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago