#159 · Primary category: Deep Learning Frameworks
EBT
PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
Project last updated:04/21/26
GitHub Stars
651
Forks
89
Contributors
4
License
Apache-2.0
Why we included this project
Energy-Based Transformers take a different route from the standard feed-forward stack: the model reasons over every token it predicts instead of emitting one forward pass. The repo backs the paper with pretraining and inference scripts for language, image, and video, plus a minimal single-file training loop that pits a Transformer++ baseline against an EBT, which is the fastest way to see the mechanics. It is research code, not a library, so expect to adapt the job scripts, wandb setup, and HPC launchers to your own cluster. If you are working on System 2 reasoning, energy-based objectives, or how alternative attention architectures scale, this is a solid place to start reproducing the paper's experiments.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.