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:
- ☆562Updated last week
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆123Updated 7 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆97Updated 3 months ago
- ☆119Updated 5 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated last month
- My take on Flow Matching☆84Updated 10 months ago
- 🪄 Interpreto is an interpretability toolbox for LLMs☆61Updated last week
- ☆14Updated 6 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆105Updated last year
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆67Updated 3 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆278Updated this week
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆75Updated 5 months ago
- Implementation of Danijar's latest iteration for his Dreamer line of work☆110Updated last week
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆117Updated 2 months ago
- σ-GPT: A New Approach to Autoregressive Models☆69Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆144Updated 6 months ago
- Clarity: A Minimalist Website Template for AI Research☆166Updated 10 months ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆113Updated last year
- ☆150Updated last year
- Explorations into improving ViTArc with Slot Attention☆43Updated last year
- The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.☆42Updated 7 months ago
- ☆73Updated last year
- ☆47Updated 3 weeks ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆55Updated 7 months ago
- Gradient Boosting Reinforcement Learning (GBRL)☆126Updated 2 weeks ago
- A natively parallel dataloader for Python, written in Rust. Serving data at GB/s speeds, while covering aspect ratio bucketing, crop and …☆119Updated 3 weeks ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆121Updated 9 months ago
- FID computation in Jax/Flax.☆29Updated last year
- A simple, performant and scalable JAX-based world modeling codebase☆111Updated 3 weeks ago
- Reliable, minimal and scalable library for pretraining foundation and world models☆90Updated this week