bhoov / hamuxLinks
Hierarchical Associative Memory User Experience
☆100Updated last year
Alternatives and similar repositories for hamux
Users that are interested in hamux are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆57Updated last year
- Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …☆72Updated 2 years ago
- Running Jax in PyTorch Lightning☆102Updated 5 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆104Updated 10 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆45Updated this week
- Code for the paper "Predictive Coding Approximates Backprop along Arbitrary Computation Graphs"☆156Updated 4 years ago
- A functional training loops library for JAX☆88Updated last year
- ☆192Updated last month
- Riemannian Optimization Using JAX☆49Updated last year
- JAX Arrays for human consumption☆91Updated last year
- ☆246Updated 3 weeks ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆151Updated 7 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆29Updated 3 years ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆30Updated last month
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆122Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- Predictive Coding JAX-based library☆61Updated last month
- Compositional Linear Algebra☆474Updated last week
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated last year
- ☆157Updated last year
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Normalizing Flows using JAX☆83Updated last year
- ☆114Updated this week
- Pytorch-like dataloaders for JAX.☆83Updated last week
- Algorithms for computations on random manifolds made easier☆90Updated last year
- Mathematical operations for JAX pytrees☆198Updated 6 months ago
- Easy Hypernetworks in Pytorch and Jax☆100Updated 2 years ago