infocusp / diffusion_modelsLinks
Minimal standalone example of diffusion model
☆159Updated 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 2 years ago
- All about the fundamentals and working of Diffusion Models☆158Updated 2 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆204Updated last year
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆150Updated 2 years ago
- Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"☆196Updated 2 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- v objective diffusion inference code for JAX.☆214Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆88Updated 2 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆254Updated 3 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆107Updated 4 years ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆123Updated 11 months ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆89Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆163Updated last year
- Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can gener…☆207Updated 2 years ago
- JAX implementation of VQGAN☆93Updated 3 years ago
- Notes and plans for fastdiffusion course☆206Updated 2 years ago
- ☆57Updated 3 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆91Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 3 years ago
- My implementation of DeepMind's Perceiver☆63Updated 4 years ago
- Easy Hypernetworks in Pytorch and Jax☆103Updated 2 years ago
- Optimized library for large-scale extraction of frames and audio from video.☆204Updated last year
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago