← all repositories

tanelp/tiny-diffusion

A minimal PyTorch implementation of denoising diffusion probabilistic models trained on 2D point datasets.

1k stars Jupyter Notebook LearningImage · Video · Audio
tiny-diffusion
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository provides a concise implementation of DDPM-style diffusion models for 2D datasets, designed as a learning resource rather than a production tool. It includes both the forward diffusion process (gradually adding noise) and the reverse denoising process (learning to recover data distributions). The README documents ablation studies on hyperparameters including learning rate, model size, number of timesteps, and variance schedules, with visualizations showing how each affects the quality of generated 2D point distributions.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.