Vrushank264 / mini-diffusionLinks
A Small diffusion model in PyTorch.
☆16Updated last year
Alternatives and similar repositories for mini-diffusion
Users that are interested in mini-diffusion are comparing it to the libraries listed below
Sorting:
- Jax/Flax implementation of Denoising Diffusion Implicit Models☆20Updated 3 years ago
- Code for "Goal-Guided Neural Cellular Automata: Learning to Control Self-Organising Systems"☆56Updated 3 years ago
- minimal diffusion model for self-study☆28Updated 2 years ago
- FID computation in Jax/Flax.☆29Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Understanding RL vision Distill article☆25Updated 2 years ago
- ☆41Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- flexible meta-learning in jax☆15Updated 2 years ago
- A minimal implementation of a denoising diffusion model in PyTorch.☆131Updated 2 weeks ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated last year
- Easy Hypernetworks in Pytorch and Jax☆106Updated 2 years ago
- ☆36Updated 3 years ago
- A Python library for Continual Inference Networks in PyTorch☆54Updated 9 months ago
- Variational Autoencoder with non-euclidean (hyperbolic) latent space☆12Updated 3 years ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Updated 5 years ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆70Updated last month
- Explorations into the proposed Streaming Deep Reinforcement Learning, from University of Alberta☆23Updated last year
- ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)☆50Updated 3 years ago
- JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.☆53Updated 3 years ago
- AdaCat☆49Updated 3 years ago
- Implementation of Diffusion Transformer Model in Pytorch☆71Updated 7 months ago
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆29Updated last year
- A JAX nn library☆22Updated 3 months ago
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 3 years ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25Updated last year
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- Cross-compilation of PyTorch armv7l (32bit) for RaspberryPi OS☆20Updated 3 years ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆23Updated 2 years ago