AlejandroBaron / tiny-diffLinks
Hackable pure-torch implementation of diffusion models
☆23Updated last year
Alternatives and similar repositories for tiny-diff
Users that are interested in tiny-diff are comparing it to the libraries listed below
Sorting:
- Implementation of Diffusion Transformer (DiT) in JAX☆306Updated last year
- All about the fundamentals and working of Diffusion Models☆159Updated 3 years ago
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆130Updated last year
- Simple and readable code for training and sampling from diffusion models☆693Updated 7 months ago
- Flow-matching algorithms in JAX☆115Updated last year
- ☆55Updated last year
- Reliable, minimal and scalable library for pretraining foundation and world models☆123Updated last week
- ☆215Updated last year
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆980Updated last year
- Code for our NeurIPS 2022 paper☆371Updated 3 years ago
- My take on Flow Matching☆90Updated last year
- Deep Learning, an Energy Approach☆241Updated 7 months ago
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 3 years ago
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Text to Image Latent Diffusion using a Transformer core☆219Updated last year
- IVON optimizer for neural networks based on variational learning.☆81Updated last year
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆156Updated 3 years ago
- Annotated version of the Mamba paper☆495Updated last year
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆352Updated 2 months ago
- Diffusion Reading Group at EleutherAI☆335Updated 2 years ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆109Updated 3 months ago
- The boundary of neural network trainability is fractal☆221Updated last year
- Train VAE like a boss☆313Updated last year
- Diffusion models in PyTorch☆124Updated last week
- Collecting research materials on neural samplers with diffusion/flow models☆59Updated 7 months ago
- A Tutorial for Diffusion Models☆60Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆161Updated 3 years ago