LouisRouss / DiffuLab
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
☆26Updated last week
Alternatives and similar repositories for DiffuLab:
Users that are interested in DiffuLab are comparing it to the libraries listed below
- Human in the loop Reinforcement Learning suite☆13Updated this week
- 🧩 Create your own puzzle, use my agents to solve it 🤖 try them out! 🧩☆9Updated 3 years ago
- ✨🌲 Hierarchical extreme multiclass and multi-label classification.☆17Updated 2 years ago
- ☆33Updated 7 months ago
- Train vision models using JAX and 🤗 transformers☆97Updated 3 weeks ago
- ☆30Updated 5 months ago
- Exploration into the Firefly algorithm in Pytorch☆38Updated 2 months ago
- Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.☆26Updated 6 months ago
- Tree-based indexes for neural-search☆31Updated last year
- Synthetic Alphabet Dataset☆18Updated last month
- Implementation of Soft Actor Critic and some of its improvements in Pytorch☆56Updated 2 months ago
- ☆53Updated last year
- Focused on fast experimentation and simplicity☆71Updated 4 months ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆50Updated last month
- ☆39Updated 3 years ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆115Updated this week
- ☆34Updated 4 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆58Updated last week
- ☆22Updated 10 months ago
- Mobile Viewer for W&B, built on top of Flutter.☆34Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆31Updated 6 months ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆14Updated last month
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆99Updated 4 months ago
- ☆52Updated 5 months ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆28Updated 4 years ago
- Explorations into improving ViTArc with Slot Attention☆40Updated 6 months ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆66Updated 8 months ago
- ☆16Updated last week
- σ-GPT: A New Approach to Autoregressive Models☆63Updated 8 months ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆27Updated 2 weeks ago