Venom: A PyTorch Generative Modeling Toolkit
For newcomers and educators, Venom reduces fragmentation in learning and comparing generative models by offering a single coherent codebase with reproducible entry points.
Venom provides a unified PyTorch toolkit implementing multiple generative modeling families (diffusion, flow matching, VAEs, normalizing flows, GANs, energy-based models) with consistent APIs and MNIST-first examples, aimed at education and prototyping rather than large-scale performance.
Modern generative modeling has grown into a broad collection of related but often separately implemented paradigms, including denoising diffusion models, score-based stochastic differential equations, flow matching, variational autoencoders, normalizing flows, adversarial models, and energy-based models. For newcomers, this fragmentation makes it difficult to compare training objectives, inference procedures, sampling algorithms, and conditioning mechanisms within a single coherent codebase. We introduce V ENOM, an educational PyTorch toolkit that implements representative generative modeling families under a unified, MNIST-first interface. V ENOM emphasizes breadth, readability, reproducible entry points, and consistent training and sampling APIs rather than large-scale performance engineering. The package currently includes diffusion and score-based models, flow matching and one-step generators, variational autoencoders, normalizing flows, generative adversarial networks, and energy-based models. It provides separate training and sampling scripts, classifier and classifier-free guidance examples, bilingual tutorial notebooks, and a model-family organization that supports teaching, prototyping, and lightweight benchmarking.