← all repositories

EugenHotaj/pytorch-generative

PyTorch library providing reference implementations of state-of-the-art generative models with training utilities.

437 stars Python ML Frameworks
pytorch-generative
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository is a Python library that makes generative modeling easier in PyTorch by providing high-quality implementations of generative models including autoregressive models (PixelCNN, PixelSNAIL, ImageGPT), variational autoencoders (VAE, VQ-VAE, VD-VAE, Beta-VAE), and normalizing flows. It offers reusable building blocks, training utilities, debugging tools, and TensorBoard integration for visualizing metrics.

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