Vector (and Scalar) Quantization, in Pytorch

Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

Official implementation of Würstchen: Efficient Pretraining of Text-to-Image Models

This repository contains implementations and illustrative code to accompany DeepMind publications

Segmind Distilled diffusion

Kandinsky 2 — multilingual text2image latent diffusion model

Huggingface-compatible SDXL Unet implementation that is readily hackable

A JAX implementation of the continuous time formulation of Consistency Models

[NeurIPS 2023] UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Models

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

v objective diffusion inference code for PyTorch.

Karras et al. (2022) diffusion models for PyTorch

Official code for "DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps" (Neurips 2022 Oral)

[ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator

Denoising Diffusion Probabilistic Models

Denoising Diffusion Implicit Models

Official repo for consistency models.

Implementation of Muse: Text-to-Image Generation via Masked Generative Transformers, in Pytorch

Consistency Distilled Diff VAE

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

High-Resolution Image Synthesis with Latent Diffusion Models

Code Repository for Liquid Time-Constant Networks (LTCs)

Training, data processing, and analysis code used for the paper "Robust Visual Flight Navigation with Liquid Neural Networks".

An adversarial example library for constructing attacks, building defenses, and benchmarking both

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention

Examples and guides for using the OpenAI API