The Energy Transformer block, in JAX
☆67Dec 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:
- Hierarchical Associative Memory User Experience☆113Jan 13, 2026Updated 4 months ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 10 months ago
- Official repository for the paper "Automating Continual Learning"☆19Jun 11, 2025Updated 11 months ago
- Repository for Sparse Universal Transformers☆20Oct 23, 2023Updated 2 years ago
- Pytorch implementation of an energy transformer - an energy-based reccurrent variant of the transformer.☆14Jul 11, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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 3 years ago
- ☆13Mar 21, 2023Updated 3 years ago
- Official code for Slot-Transformer for Videos (STEVE)☆51Jan 9, 2023Updated 3 years ago
- This is the official repository for the paper "Laplacian Features for Learning with Hyperbolic Space"☆14Aug 8, 2022Updated 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
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14May 26, 2025Updated last year
- ☆26Dec 3, 2023Updated 2 years ago
- program synthesis with neuro-symbolic differentiable interpreters☆17Sep 17, 2025Updated 8 months ago
- ☆18Mar 9, 2023Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 6 months ago
- A generative model of compositionality in symmetric monoidal (Kleisli) categories☆12Oct 4, 2023Updated 2 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☆16Nov 24, 2025Updated 6 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆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
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Apr 8, 2022Updated 4 years ago
- ☆20Apr 3, 2023Updated 3 years ago
- ☆18Nov 10, 2023Updated 2 years ago
- SWIM-IR is a Synthetic Wikipedia-based Multilingual Information Retrieval training set with 28 million query-passage pairs spanning 33 la…☆49Nov 13, 2023Updated 2 years ago
- Example application of the paper "A Neuro-vector-symbolic architecture for Solving Raven's Progressive Matrices" published in Nature Mach…☆33Sep 17, 2025Updated 8 months ago
- ☆12Jan 4, 2024Updated 2 years ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆57May 17, 2024Updated 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.
- Code for verifying deep neural feature ansatz☆22May 3, 2023Updated 3 years ago
- ☆12Sep 26, 2019Updated 6 years ago
- Code for reproducing the paper "Neural Networks Fail to Learn Periodic Functions and How to Fix It" as part of the ML Reproducibility Cha…☆11Apr 16, 2021Updated 5 years ago
- ☆16Jun 13, 2023Updated 2 years ago
- Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …☆15Mar 11, 2024Updated 2 years ago
- Online inference on reactive probabilistic models, with SMC and symbolic methods☆16Jun 7, 2023Updated 2 years ago
- ☆14Mar 31, 2024Updated 2 years ago