PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
☆665Apr 21, 2026Updated 6 months ago
Alternatives and similar repositories for EBT
Users that are interested in EBT are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆99Jun 14, 2024Updated 2 years ago
- minimal Energy-based transformer☆43Dec 11, 2025Updated 10 months ago
- PyTorch Code for Explorative Modeling: Unlocking a Third Pretraining Axis and End-to-End Generation☆253Sep 30, 2026Updated last month
- Universal Reasoning Model☆138Jan 15, 2026Updated 9 months ago
- Continuous Thought Machines, because thought takes time and reasoning is a process.☆2,045Dec 29, 2025Updated 10 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆221Oct 9, 2025Updated last year
- 🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffus…☆117Updated this week
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆27Jan 28, 2024Updated 2 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆50Feb 27, 2023Updated 3 years ago
- Official JAX implementation of End-to-End Test-Time Training for Long Context☆717Feb 15, 2026Updated 8 months ago
- Code for BLT research paper☆2,070Nov 3, 2025Updated last year
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,307Jul 6, 2026Updated 4 months ago
- [NeurIPS'25] Generalizable Reasoning through Compositional Energy Minimization☆29Oct 28, 2025Updated last year
- H-Net: Hierarchical Network with Dynamic Chunking☆926Nov 20, 2025Updated 11 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Official implementation of Categorical Flow Maps on text.☆70Feb 16, 2026Updated 8 months ago
- PyTorch code and models for VJEPA2 self-supervised learning from video.☆4,860Oct 29, 2026Updated last week
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆1,042Jul 10, 2025Updated last year
- dLLM: Simple Diffusion Language Modeling☆2,727Jul 17, 2026Updated 3 months ago
- Official PyTorch implementation for "Large Language Diffusion Models"☆4,015Jul 15, 2026Updated 3 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆748Feb 29, 2024Updated 2 years ago
- Flash Attention Triton kernel with support for second-order derivatives☆183Oct 2, 2026Updated last month
- ☆1,407Jan 25, 2026Updated 9 months ago
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆229Jul 31, 2026Updated 3 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.
- Jax Codebase for Evolutionary Strategies at the Hyperscale☆385Feb 27, 2026Updated 8 months ago
- The code repository of the paper: Competition and Attraction Improve Model Fusion☆170Aug 25, 2025Updated last year
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆275Jun 3, 2026Updated 5 months ago
- Software for using predictive coding algorithms to train PyTorch models.☆30Feb 13, 2024Updated 2 years ago
- Pretraining and inference code for a large-scale depth-recurrent language model☆991Dec 29, 2025Updated 10 months ago
- An open source library designed to provide community examples of Joint Embedding Predictive Architectures (JEPAs). It contains code and e…☆807Jul 17, 2026Updated 3 months ago
- Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computation (NeurIPS 2025)☆649Sep 19, 2026Updated last month
- ☆56Oct 20, 2026Updated 2 weeks ago
- PyTorch code and models for V-JEPA self-supervised learning from video.☆4,191Feb 27, 2025Updated last year
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- [NeurIPS 2024] Simple and Effective Masked Diffusion Language Model☆724Sep 29, 2025Updated last year
- A platform for reproducible world model research and evaluation☆2,298Oct 2, 2026Updated last month
- Evolution Pretraining Fully in Int Formats☆189Feb 25, 2026Updated 8 months ago
- Unofficial implementation of Titans, SOTA memory for transformers, in Pytorch☆2,004Updated this week
- ☆6,550Apr 1, 2026Updated 7 months ago
- Cambrian-S: Towards Spatial Supersensing in Video☆573Apr 3, 2026Updated 7 months ago
- Research code artifacts for Code World Model (CWM) including inference tools, reproducibility, and documentation.☆917Jul 17, 2026Updated 3 months ago