LouisRouss / DiffuLabLinks
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
☆42Updated 3 weeks ago
Alternatives and similar repositories for DiffuLab
Users that are interested in DiffuLab are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 4 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆130Updated 3 weeks ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Updated 6 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆72Updated 2 months ago
- Explorations into improving ViTArc with Slot Attention☆43Updated last year
- Implementation of Danijar's latest iteration for his Dreamer line of work☆161Updated last week
- ☆123Updated 7 months ago
- ☆860Updated last week
- Gradient Boosting Reinforcement Learning (GBRL)☆136Updated this week
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆153Updated 9 months ago
- ☆14Updated 9 months ago
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆198Updated 3 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆126Updated last year
- Flash Attention Triton kernel with support for second-order derivatives☆138Updated last month
- Focused on fast experimentation and simplicity☆80Updated last year
- Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"☆182Updated last year
- Clarity: A Minimalist Website Template for AI Research☆179Updated last year
- My take on Flow Matching☆90Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆129Updated 3 weeks ago
- FID computation in Jax/Flax.☆29Updated last year
- An open source library designed to provide community examples of Joint Embedding Predictive Architectures (JEPAs). It contains code and e…☆46Updated 2 weeks ago
- Implementation of Diffusion Transformer Model in Pytorch☆74Updated last month
- Implementation of a transformer for reinforcement learning using `x-transformers`☆72Updated 4 months ago
- Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".☆144Updated 2 months ago
- Induce brain-like topographic structure in your neural networks☆71Updated 6 months ago
- [ICML-2025] We introduce Lie group Relative position Encodings (LieRE) that goes beyond RoPE in supporting n-dimensional inputs.☆28Updated 5 months ago
- Reliable, minimal and scalable library for evaluate and conduct world model research☆49Updated this week
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated last year
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆55Updated 10 months ago