DruvPai / DiffusionLabLinks
Easy no-frills Jax implementations of common abstractions for simple diffusion models.
☆10Updated 9 months ago
Alternatives and similar repositories for DiffusionLab
Users that are interested in DiffusionLab are comparing it to the libraries listed below
Sorting:
- ☆851Updated last week
- PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning☆584Updated 2 months ago
- This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"☆216Updated 11 months ago
- ☆658Updated 9 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 4 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆698Updated last year
- A curated list for awesome discrete diffusion models resources.☆531Updated 4 months ago
- code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆1,144Updated 2 months ago
- My take on Flow Matching☆90Updated last year
- ☆83Updated last year
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆255Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆107Updated 3 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆130Updated 2 weeks ago
- Official implementation of Inductive Moment Matching☆570Updated 6 months ago
- [NeurIPS 2024] Simple and Effective Masked Diffusion Language Model☆610Updated 4 months ago
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆84Updated 8 months ago
- Unofficial implementation of the toy example in JiT https://arxiv.org/abs/2511.13720☆57Updated 2 months ago
- Interactive visualizations of the geometric intuition behind diffusion models.☆1,054Updated this week
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆29Updated 7 months ago
- Reading list for research topics in state-space models☆341Updated 7 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆305Updated last year
- ☆729Updated last year
- Minimal Implementation of a D3PM in pytorch☆285Updated last year
- This is the code repository for IntPhys 2, a video benchmark designed to evaluate the intuitive physics understanding of deep learning mo…☆93Updated 3 months ago
- A framework for training world models with virtual environments, complete with annotated environment dataset (RetroAct), exploration agen…☆69Updated 3 months ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆170Updated last year
- Implementation of a framework for Genie2 in Pytorch☆156Updated last year
- Code release for paper "Test-Time Training Done Right"☆367Updated 3 weeks ago
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆949Updated 6 months ago
- Annotated Flow Matching paper☆227Updated last year