infocusp / diffusion_modelsLinks
Minimal standalone example of diffusion model
☆163Updated 3 years ago
Alternatives and similar repositories for diffusion_models
Users that are interested in diffusion_models are comparing it to the libraries listed below
Sorting:
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 3 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆206Updated 2 years ago
- All about the fundamentals and working of Diffusion Models☆159Updated 3 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆156Updated 3 years ago
- Contrastive Language-Image Pretraining☆144Updated 3 years ago
- An alternative to convolution in neural networks☆259Updated last year
- v objective diffusion inference code for JAX.☆214Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 4 years ago
- Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"☆198Updated 3 years ago
- Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can gener…☆205Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆108Updated 4 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- JAX implementation of VQGAN☆91Updated 3 years ago
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆94Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago
- Easy Hypernetworks in Pytorch and Jax☆106Updated 3 years ago
- ☆58Updated 3 years ago
- Pedagogical codebase for a simplified score-based generative model design, with training loop☆40Updated 4 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆190Updated 3 years ago
- My implementation of DeepMind's Perceiver☆63Updated 4 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- ☆75Updated 3 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆265Updated 10 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆476Updated 3 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆89Updated 4 years ago
- A GPT, made only of MLPs, in Jax☆59Updated 4 years ago
- Optimized library for large-scale extraction of frames and audio from video.☆204Updated 2 years ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆207Updated last year
- ☆160Updated 3 years ago