🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"

A latent text-to-image diffusion model

Masked Visual Pre-training for Robotics

Official PyTorch Implementation of "GAN-Supervised Dense Visual Alignment" (CVPR 2022 Oral, Best Paper Finalist)

GLIDE: a diffusion-based text-conditional image synthesis model

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

Official Implementation of Swapping Autoencoder for Deep Image Manipulation (NeurIPS 2020)

Official Implementation of "The Hessian Penalty: A Weak Prior for Unsupervised Disentanglement" (ECCV 2020 Spotlight)

Synchronized Batch Normalization implementation in PyTorch.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Datasets, Transforms and Models specific to Computer Vision