LouisRouss / DiffuLabLinks
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
☆28Updated this week
Alternatives and similar repositories for DiffuLab
Users that are interested in DiffuLab are comparing it to the libraries listed below
Sorting:
- ☆14Updated 2 months ago
- Focused on fast experimentation and simplicity☆76Updated 6 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆101Updated 8 months ago
- My take on Flow Matching☆66Updated 6 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆97Updated last week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆92Updated 3 months ago
- ☆110Updated last month
- Explorations into improving ViTArc with Slot Attention☆42Updated 8 months ago
- FID computation in Jax/Flax.☆28Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆128Updated 2 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆251Updated this week
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆101Updated 6 months ago
- ☆33Updated 6 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated 11 months ago
- ☆11Updated 7 months ago
- Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.☆103Updated this week
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated last month
- Exploration into the Firefly algorithm in Pytorch☆40Updated 5 months ago
- Gradient Boosting Reinforcement Learning (GBRL)☆116Updated last month
- ☆23Updated last year
- ☆53Updated last year
- Lightweight package that tracks and summarizes code changes using LLMs (Large Language Models)☆34Updated 4 months ago
- ☆150Updated 11 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆65Updated last month
- Train vision models using JAX and 🤗 transformers☆98Updated 3 months ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 8 months ago
- Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.☆114Updated last month
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 11 months ago
- RS-IMLE☆41Updated 7 months ago
- ☆24Updated 2 months ago