The Energy Transformer block, in JAX
☆68Dec 14, 2023Updated 2 years ago
Alternatives and similar repositories for energy-transformer-jax
Users that are interested in energy-transformer-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Official Implementation of Energy Transformer in PyTorch for Mask Image Reconstruction☆32Mar 10, 2024Updated 2 years ago
- Hierarchical Associative Memory User Experience☆113Jan 13, 2026Updated 5 months ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 11 months ago
- Official repository for the paper "Automating Continual Learning"☆20Jun 11, 2025Updated last year
- Repository for Sparse Universal Transformers☆20Oct 23, 2023Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Official Code Repository for the paper "Key-value memory in the brain"☆32Feb 25, 2025Updated last year
- Pytorch implementation of an energy transformer - an energy-based reccurrent variant of the transformer.☆14Jul 11, 2023Updated 2 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆17Nov 8, 2024Updated last year
- repo for code for paper on general theory associative memory models☆22Jun 15, 2022Updated 4 years ago
- ☆13Mar 21, 2023Updated 3 years ago
- ☆10May 24, 2021Updated 5 years ago
- the reflective tower Blond by Olivier Danvy & Karoline Malmkjær☆16May 21, 2025Updated last year
- ☆17Mar 2, 2023Updated 3 years ago
- ☆37Oct 13, 2021Updated 4 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆11Dec 30, 2024Updated last year
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14May 26, 2025Updated last year
- ☆27Dec 3, 2023Updated 2 years ago
- ☆18Mar 9, 2023Updated 3 years ago
- A generative model of compositionality in symmetric monoidal (Kleisli) categories☆12Oct 4, 2023Updated 2 years ago
- 3D geoms for plotnine (grammar of graphics in Python)☆13Aug 5, 2022Updated 3 years ago
- First-order logic theorem prover supporting unification with approximate vector similarity☆14Mar 23, 2023Updated 3 years ago
- General framework for Bayesian inversion of continuous hierarchical models☆10Sep 20, 2021Updated 4 years ago
- A flat container abstraction for Rust☆17Nov 24, 2025Updated 6 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Create embeddings for LLM using the Nomic API☆23Nov 21, 2024Updated last year
- Implementing scalable LLMs in pure JAX (no third-party libraries)☆50Jun 11, 2026Updated last week
- ☆13Nov 13, 2020Updated 5 years ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆38Aug 27, 2024Updated last year
- ☆20Apr 3, 2023Updated 3 years ago
- ☆19Mar 25, 2025Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆57Dec 4, 2024Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Mar 24, 2023Updated 3 years ago
- ☆19Nov 10, 2023Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆12Jan 4, 2024Updated 2 years ago
- Explorations into adversarial losses on top of autoregressive loss for language modeling☆41Dec 21, 2025Updated 5 months ago
- Python package for Geometric / Clifford Algebra with Pytorch.☆15Jun 2, 2026Updated 2 weeks ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆57May 17, 2024Updated 2 years ago
- Pragmatic models for generating and following instructions☆13Dec 22, 2019Updated 6 years ago
- ☆12Sep 26, 2019Updated 6 years ago
- Code for verifying deep neural feature ansatz☆22May 3, 2023Updated 3 years ago