LouisRouss / DiffuLabLinks
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
☆30Updated last week
Alternatives and similar repositories for DiffuLab
Users that are interested in DiffuLab are comparing it to the libraries listed below
Sorting:
- ☆14Updated 3 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆103Updated 9 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆97Updated 3 weeks ago
- Focused on fast experimentation and simplicity☆75Updated 8 months ago
- 🐟 CODS - Conformal Object Detection and Segmentation☆16Updated last month
- An implementation of PSGD Kron second-order optimizer for PyTorch☆96Updated last month
- ☆33Updated 7 months ago
- Gradient Boosting Reinforcement Learning (GBRL)☆118Updated 2 weeks ago
- ☆150Updated last year
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆65Updated 3 weeks ago
- ☆115Updated 2 months ago
- Getting crystal-like representations with harmonic loss☆194Updated 5 months ago
- 🪄 Interpreto is an interpretability toolbox for LLMs☆34Updated this week
- σ-GPT: A New Approach to Autoregressive Models☆67Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆124Updated last year
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆53Updated 5 months ago
- ☆53Updated last year
- Exploration into the Firefly algorithm in Pytorch☆40Updated 6 months ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆116Updated 4 months ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆35Updated 3 weeks ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆265Updated this week
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆130Updated 4 months ago
- A simple, performant and scalable JAX-based world modeling codebase☆70Updated last week
- ☆53Updated 9 months ago
- BricksRL: A Platform for Democratizing Robotics and Reinforcement Learning Research and Education with LEGO☆62Updated 10 months ago
- ☆23Updated last year
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆18Updated 9 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- Efficient optimizers☆256Updated 3 weeks ago
- Explorations into improving ViTArc with Slot Attention☆42Updated 10 months ago