The Energy Transformer block, in JAX
☆69Dec 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:
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆27Jan 28, 2024Updated 2 years ago
- Bayesian model reduction for probabilistic machine learning☆12Jul 3, 2025Updated last year
- Hierarchical Associative Memory User Experience☆112Jan 13, 2026Updated 9 months ago
- Official repository for the paper "Automating Continual Learning"☆21Jun 11, 2025Updated last year
- Very minimal implementation of DEER☆14Jul 6, 2026Updated 3 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Pytorch implementation of an energy transformer - an energy-based reccurrent variant of the transformer.☆17Jul 11, 2023Updated 3 years ago
- Official Code Repository for the paper "Key-value memory in the brain"☆34Feb 25, 2025Updated last year
- ☆13Mar 21, 2023Updated 3 years ago
- Code for the paper Depth Memory☆14Oct 14, 2026Updated 2 weeks ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- ☆10May 24, 2021Updated 5 years ago
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 7 months ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆23Nov 8, 2024Updated last year
- This is the official repository for the paper "Laplacian Features for Learning with Hyperbolic Space"☆14Aug 8, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Based on lectures on the LLM-Verifier Interface at the Summer School on Foundations of Programming and Software Systems (FoPSS 2026)☆17Jul 14, 2026Updated 3 months ago
- the reflective tower Blond by Olivier Danvy & Karoline Malmkjær☆16May 21, 2025Updated last year
- ☆27Dec 3, 2023Updated 2 years ago
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 7 months ago
- ☆20Mar 25, 2025Updated last year
- A flat container abstraction for Rust☆17Nov 24, 2025Updated 11 months ago
- program synthesis with neuro-symbolic differentiable interpreters☆17Sep 17, 2025Updated last year
- ☆16Jan 26, 2023Updated 3 years ago
- [ESWA 2023] Pytorch Implementation for 'Orientation and Context Entangled Network for Retinal Vessel Segmentation' Expert Systems with A…☆13May 13, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆32May 30, 2024Updated 2 years ago
- A dataset of segmented RNA molecule surfaces, as a benchmark task in 3D machine learning on surfaces. From Poulenard et al., 3DV 2019.☆12Sep 27, 2021Updated 5 years ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆37Oct 3, 2025Updated last year
- Official code implementation for the paper "Do Vision & Language Decoders use Images and Text equally? How Self-consistent are their Expl…☆12Jul 14, 2026Updated 3 months ago
- Easy bindings between Lean and Python.☆37Updated this week
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆70Apr 8, 2022Updated 4 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆48Mar 24, 2023Updated 3 years ago
- Various utilities for the Brevitas (https://github.com/Xilinx/brevitas) framework for neural network quantization in PyTorch☆11Jan 16, 2026Updated 9 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- First-order logic theorem prover supporting unification with approximate vector similarity☆14Mar 23, 2023Updated 3 years ago
- Code release for ICLR 2023 paper: SlotFormer on object-centric dynamics models☆127Sep 20, 2023Updated 3 years ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆38Aug 27, 2024Updated 2 years ago
- Example application of the paper "A Neuro-vector-symbolic architecture for Solving Raven's Progressive Matrices" published in Nature Mach…☆34Sep 17, 2025Updated last year
- ☆12Sep 26, 2019Updated 7 years ago
- 3D geoms for plotnine (grammar of graphics in Python)☆14Aug 5, 2022Updated 4 years ago
- Code for "Long Sequence Hopfield Memory"☆11Apr 23, 2024Updated 2 years ago