LouisRouss / DiffuLabLinks
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
☆40Updated last week
Alternatives and similar repositories for DiffuLab
Users that are interested in DiffuLab are comparing it to the libraries listed below
Sorting:
- ☆122Updated 6 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Updated last year
- ☆33Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆148Updated 8 months ago
- RS-IMLE☆43Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆97Updated 5 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆70Updated last month
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 3 months ago
- σ-GPT: A New Approach to Autoregressive Models☆70Updated last year
- My take on Flow Matching☆89Updated 11 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆116Updated last year
- Implementation of Danijar's latest iteration for his Dreamer line of work☆150Updated this week
- Clarity: A Minimalist Website Template for AI Research☆173Updated 11 months ago
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆192Updated 2 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆292Updated this week
- Flash Attention Triton kernel with support for second-order derivatives☆129Updated 2 weeks ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆125Updated 3 weeks ago
- A simple, performant and scalable JAX-based world modeling codebase.☆119Updated 2 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 11 months ago
- ☆82Updated last year
- Induce brain-like topographic structure in your neural networks☆71Updated 5 months ago
- ☆30Updated last year
- Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).☆67Updated 3 months ago
- Focused on fast experimentation and simplicity☆79Updated last year
- [ICML 2025] Implementation of Spatial Reasoning with Denoising Models☆85Updated 5 months ago
- Explorations into improving ViTArc with Slot Attention☆43Updated last year
- The official Python SDK for the Perceptron API☆58Updated this week
- ☆172Updated 3 months ago
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆121Updated last month
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆97Updated 11 months ago