pybetter

Tool for fixing trivial problems with your code.


lapce

Lightning-fast and Powerful Code Editor written in Rust


TokenMix

TokenMix: Rethinking Image Mixing for Data Augmentation in Vision Transformers (ECCV 2022)


optimum

🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools


awesome-ai-art-image-synthesis

A list of awesome tools, ideas, prompt engineering tools, colabs, models, and helpers for the prompt designer playing with aiArt and image synthesis. Covers Dalle2, MidJourney, StableDiffusion, and open source tools.


stable-diffusion-webui

Stable Diffusion web UI


google-images-download

Python Script to download hundreds of images from 'Google Images'. It is a ready-to-run code!


Awesome-Image-Harmonization

A curated list of papers, code and resources pertaining to image harmonization.


discord-stable-diffusion

A neat Discord bot to run Stable Diffusion locally


[SIGGRAPH Asia 2022] IDE-3D: Interactive Disentangled Editing For High-Resolution 3D-aware Portrait Synthesis


coyo-dataset

COYO-700M: Large-scale Image-Text Pair Dataset


piq

Measures and metrics for image2image tasks. PyTorch.


Awesome-Image-Composition

A curated list of papers, code and resources pertaining to image composition/compositing, which aims to generate realistic composite image.


A port of GOBIG for Stable Diffusion


Code behind the work "Single Cortical Neurons as Deep Artificial Neural Networks", published in Neuron 2021


Visual Taste Approximator (VTA) is a very simple tool that helps anyone create an automatic replica of themselves that can approximate their own personal visual taste


gpu-tester

gpu tester detects broken and slow gpus in a cluster


audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch


mmcv

OpenMMLab Computer Vision Foundation


mmdetection

OpenMMLab Detection Toolbox and Benchmark


OV-DETR

[Under preparation] Code repo for "Open-Vocabulary DETR with Conditional Matching" (ECCV 2022)


DAB-DETR

[ICLR 2022] Official implementation of the paper "DAB-DETR: Dynamic Anchor Boxes are Better Queries for DETR"


DN-DETR

[CVPR 2022 Oral] Official implementation of DN-DETR


DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"


MaskDINO

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"


zero-shot-image-to-text

Implementation of Zero-Shot Image-to-Text Generation for Visual-Semantic Arithmetic


distributed-pytorch-template

This is a seed project for distributed PyTorch training, which was built to customize your network quickly


Palette-Image-to-Image-Diffusion-Models

Unofficial implementation of Palette: Image-to-Image Diffusion Models by Pytorch


Style-Your-Hair

Official Pytorch implementation of "Style Your Hair: Latent Optimization for Pose-Invariant Hairstyle Transfer via Local-Style-Aware Hair Alignment (ECCV 2022)"


model-uncertainty-for-adaptation

Code paper Uncertainty Reduction for Uncertainty Reduction for Model Adaptation in Semantic Segmentation at CVPR 2021


fullgrad-saliency

Full-gradient saliency maps


ttgo

A PyTorch implementation of TTGO algorithm and the applications presented in the paper "Tensor Train for Global Optimization Problems in Robotics"


GeoNeRF

Generalizing NeRF with Geometry Priors


fast-transformers

Pytorch library for fast transformer implementations


stable-diffusion

A latent text-to-image diffusion model


CrossAttentionControl

Unofficial implementation of "Prompt-to-Prompt Image Editing with Cross Attention Control" with Stable Diffusion


hickle

a HDF5-based python pickle replacement


chirpycardinal

Stanford's Alexa Prize socialbot


montlake

Montlake contains tools for geometric data analysis. It includes vector field group lasso and basis pursuit methods for parametric manifold learning. It also contains differentiable shape featurizations including interpoint distances, planar angles, and torsions from data positions for shape space analysis.


stable-diffusion-vulkan

Trying and failing to run Stable Diffusion on Vulkan


stable-diffusion-rocm-docker

Stable Diffusion Docker image preconfigured for usage with AMD Radeon cards


Stable Diffusion install/uninstall/prompt for MacOS M1


StableDiffusionTelegram

StableDiffusionTelegram is a telegram bot that allows to generate images using the stable diffusion AI from a telegram bot, in a much more comfortable and simple way.


stable-diffusion-walk

Stable Diffusion Walk with Docker environment


⚡Instant Stable Diffusion on k8s(Kubernetes) with Helm


solvent

[WIP] - AI-Assisted Texture Generation Toolkit in Blender


digil-waifu-diffusion

Generating digital images from natural language descriptions.


sdui

Local ImGui UI for Stable Diffusion. Features embedded PNG metadata, Apple M1 fixes, result caching, img2img, and more!


sd-dreamer

Qt based Linux/Windows GUI for Stable Diffusion


stable-diffusion-colab

Colab notebook for Stable Diffusion XL Turbo.


stable-dreambooth

Dreambooth implementation based on Stable Diffusion with minimal code.


deforum-stable-diffusion

Refactor of the Deforum Stable Diffusion notebook (featuring video_init) https://colab.research.google.com/github/deforum/stable-diffusion/blob/main/Deforum_Stable_Diffusion.ipynb


g-diffuser-bot

Discord bot and Interface for Stable Diffusion


stable-diffusion-nvidia-docker

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.


stable-diffusion-docker

Run the official Stable Diffusion releases in a Docker container with txt2img, img2img, depth2img, pix2pix, upscale4x, and inpaint.


Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can generate and then later interpolate between the images of your choice.


awesome-stable-diffusion

Curated list of awesome resources for the Stable Diffusion AI Model.


stable-diffusion-webui-docker

Easy Docker setup for Stable Diffusion with user-friendly UI


ai-art-generator

For automating the creation of large batches of AI-generated artwork locally.


stable-diffusion-videos

Create 🔥 videos with Stable Diffusion by exploring the latent space and morphing between text prompts


Tileable Stable Diffusion - Cog model


Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion


diffusion-ui

Frontend for deeplearning Image generation


japanese-stable-diffusion

Japanese Stable Diffusion is a Japanese specific latent text-to-image diffusion model capable of generating photo-realistic images given any text input.


immich

Self-hosted photo and video backup solution directly from your mobile phone.


Long-form text-to-images generation, using a pipeline of deep generative models (GPT-3 and Stable Diffusion)


stable-diffusion

Optimized Stable Diffusion modified to run on lower GPU VRAM


CCLM

Cross-View Language Modeling: Towards Unified Cross-Lingual Cross-Modal Pre-training (ACL 2023))


X-VLM

X-VLM: Multi-Grained Vision Language Pre-Training (ICML 2022)


SeqFormer

SeqFormer: Sequential Transformer for Video Instance Segmentation (ECCV 2022 Oral)


CRIS.pytorch

An official PyTorch implementation of the CRIS paper


PatchDiffusion-Pytorch

Code for the Paper "Improving Diffusion Model Efficiency Through Patching"


PrefixTuning

Prefix-Tuning: Optimizing Continuous Prompts for Generation


PeRFception

[NeurIPS2022] Official implementation of PeRFception: Perception using Radiance Fields.


Cozy-Auto-Texture

A Blender add-on for generating free textures using the Stable Diffusion AI text to image model.


StablenderDiffusion

General Repo for blender and stable-diffusion integration


book

The Rust Programming Language


pyannotate

Auto-generate PEP-484 annotations


flaxify

Convert haiku modules to flax


jax-hypernetwork

A simple hypernetwork implementation in jax using haiku.


ILVR: Conditioning Method for Denoising Diffusion Probabilistic Models (ICCV 2021 Oral)


Open-Vocabulary Instance Segmentation via Robust Cross-Modal Pseudo-Labeling @ CVPR22


kaolin-wisp

NVIDIA Kaolin Wisp is a PyTorch library powered by NVIDIA Kaolin Core to work with neural fields (including NeRFs, NGLOD, instant-ngp and VQAD).


XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model


cog-stable-diffusion

Diffusers Stable Diffusion as a Cog model


progrock-stable

Stable Diffusion with some Proggy Enhancements


sd-akashic

A compendium of informations regarding Stable Diffusion (SD)


aquila

An easy to use Neural Search Engine. Index latent vectors along with JSON metadata and do efficient k-NN search.


clize

CLIze: Turn Python functions into command-line interfaces


patch-torch-save

Patches the torch.save function with arbitrary code that gets executed upon torch.load.


koi

A plug-in for Krita that enables the use of AI models for img2img generation.


exporters

Export Hugging Face models to Core ML and TensorFlow Lite


diffusers-interpret

Diffusers-Interpret 🤗🧨🕵♀: Model explainability for 🤗 Diffusers. Get explanations for your generated images.


coveragepy

The code coverage tool for Python


mcpyrate

Advanced macro expander and language lab for Python.


multimethod

Multiple argument dispatching.


awkward

Manipulate JSON-like data with NumPy-like idioms.


commitizen

Create committing rules for projects :rocket: auto bump versions :arrow_up: and auto changelog generation :open_file_folder:


krita

Krita is a free and open source cross-platform application that offers an end-to-end solution for creating digital art files from scratch built on the KDE and Qt frameworks.


minicon

Minimization of the filesystem for containers


bottleneck

Fast NumPy array functions written in C


github-markdown-toc

Easy TOC creation for GitHub README.md


sygil-webui

Stable Diffusion web UI


seam-carving

A fast Python implementation of Seam Carving for Content-Aware Image Resizing.


stability-sdk

SDK for interacting with stability.ai APIs (e.g. stable diffusion inference)


NeuralPassthrough

This repository provides source code, trained neural network model and dataset for our NeuralPassthrough work that is published at SIGGRAPH 2022.


Repo for our ECCV 2022 paper on "Paint2Pix: Interactive Painting based Progressive Image Synthesis and Editing"


Cold-Diffusion-Models

Official implementation of Cold-Diffusion for different transformations in pytorch.


Disco Diffusion inpainting mode colab for animation_mode=='None'/'2D'/'3D'. For still image inpainting


auraloss

Collection of audio-focused loss functions in PyTorch


neural-deferred-shading

Multi-View Mesh Reconstruction with Neural Deferred Shading (CVPR 2022)


bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.


FGT

[ECCV 2022] Flow-Guided Transformer for Video Inpainting


musika

Fast Infinite Waveform Music Generation


steerable-nafx

Steerable discovery of neural audio effects


RAVE

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder


Simple prompt generator for Midjourney, DALLe, Stable and Disco Diffusion, and etc.


ipyvizzu-story

Build, present and share animated data stories in Jupyter Notebook and similar environments.


audio-diffusion-pytorch

Audio generation using diffusion models, in PyTorch.


FasterTransformer

Transformer related optimization, including BERT, GPT


stitching

A Python package for fast and robust Image Stitching


pugsql

A HugSQL-inspired database library for Python


OpenPSG

Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22


outline

The fastest knowledge base for growing teams. Beautiful, realtime collaborative, feature packed, and markdown compatible.


leon

🧠 Leon is your open-source personal assistant.


Offical code for the paper: "Growing 3D Artefacts and Functional Machines with Neural Cellular Automata" https://arxiv.org/abs/2103.08737


jax-nca

Neural Cellular Automata implemented with Jax


controllable-ncas

Code for "Goal-Guided Neural Cellular Automata: Learning to Control Self-Organising Systems"


hyper-nn

Easy Hypernetworks in Pytorch and Jax


evotorch

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.


bundle-adjusting-NeRF

BARF: Bundle-Adjusting Neural Radiance Fields 🤮 (ICCV 2021 oral)


mal

mal - Make a Lisp


VNext

Next-generation Video instance recognition framework on top of Detectron2 which supports InstMove (CVPR 2023), SeqFormer(ECCV Oral),  and IDOL(ECCV Oral))


Unicorn

[ECCV'22 Oral] Towards Grand Unification of Object Tracking


pyFileFixity

📂🛡Suite of tools for file fixity (data protection for long term storage⌛) using redundant error correcting codes, hash auditing and duplications with majority vote, all in pure Python🐍


ConSinGAN

PyTorch implementation of "Improved Techniques for Training Single-Image GANs" (WACV-21)


DoubleDIP

Official implementation of the paper "Double-DIP: Unsupervised Image Decomposition via Coupled Deep-Image-Priors"


Official Pytorch Implementation for "Text2LIVE: Text-Driven Layered Image and Video Editing" (ECCV 2022 Oral)


clip-jax

Train vision models using JAX and 🤗 transformers


fauxpilot

FauxPilot - an open-source alternative to GitHub Copilot server


swarmkit

A toolkit for orchestrating distributed systems at any scale. It includes primitives for node discovery, raft-based consensus, task scheduling and more.


qna

[CVPR2022 - Oral] Official Jax Implementation of Learned Queries for Efficient Local Attention


Official Implementation of "Third Time's the Charm? Image and Video Editing with StyleGAN3" (AIM ECCVW 2022) https://arxiv.org/abs/2201.13433


euporie

Jupyter notebooks in the terminal


asv

Airspeed Velocity: A simple Python benchmarking tool with web-based reporting


dataklasses

A different spin on dataclasses.


zarr-python

An implementation of chunked, compressed, N-dimensional arrays for Python.


arviz

Exploratory analysis of Bayesian models with Python


cachetools

Extensible memoizing collections and decorators


napari

napari: a fast, interactive, multi-dimensional image viewer for python


Robyn

Robyn is a High-Performance, Community-Driven, and Innovator Friendly Web Framework with a Rust runtime.


numexpr

Fast numerical array expression evaluator for Python, NumPy, Pandas, PyTables and more


aiofiles

File support for asyncio


pexpect

A Python module for controlling interactive programs in a pseudo-terminal


nbdime

Tools for diffing and merging of Jupyter notebooks.


xarray

N-D labeled arrays and datasets in Python


mercure

An open, easy, fast, reliable and battery-efficient solution for real-time communications


ultrajson

Ultra fast JSON decoder and encoder written in C with Python bindings


curio

Good Curio!


meson

The Meson Build System


twisted

Event-driven networking engine written in Python.


watchdog

Python library and shell utilities to monitor filesystem events.


RustPython

A Python Interpreter written in Rust


celery

Distributed Task Queue (development branch)


tornado

Tornado is a Python web framework and asynchronous networking library, originally developed at FriendFeed.


xstate

Actor-based state management & orchestration for complex app logic.


tabby

A terminal for a more modern age


drjit

Dr.Jit — A Just-In-Time-Compiler for Differentiable Rendering


hatch

Modern, extensible Python project management


Silice

Silice is an open source language that simplifies prototyping and writing algorithms on FPGA architectures.


learn-fpga

Learning FPGA, yosys, nextpnr, and RISC-V


distributed

A distributed task scheduler for Dask


ipdb

Integration of IPython pdb


flit

Simplified packaging of Python modules


pipx

Install and Run Python Applications in Isolated Environments


tmux

tmux source code


jupyverse

A Jupyter server based on FastAPI :rocket:


blackjax

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.


ibis

The flexibility of Python with the scale and performance of modern SQL.


pyparsing

Python library for creating PEG parsers


jinja

A very fast and expressive template engine.


mako

Mako Templates for Python


slim

Slim(toolkit): Don't change anything in your container image and minify it by up to 30x (and for compiled languages even more) making it secure too! (free and open source)


Video-Pre-Training

Video PreTraining (VPT): Learning to Act by Watching Unlabeled Online Videos


truss

The simplest way to serve AI/ML models in production


eden

Eden converts your python function into a hosted endpoint with minimal changes to your existing code :mage_man:


This is a implementation of the 3D FLAME model in PyTorch


Official PyTorch implementation of StyleGAN3


Modifications of the official PyTorch implementation of StyleGAN3. Let's easily generate images and videos with StyleGAN2/2-ADA/3!


instant-ngp-pytorch

Study for Instant neural graphics primitives (Unofficial)


denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Models in PyTorch


blur-diffusion

Official PyTorch implementation of the paper Progressive Deblurring of Diffusion Models for Coarse-to-Fine Image Synthesis.


DeepSpeed-MII

MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.


SemanticStyleGAN

release code for SemanticStyleGAN (CVPR 2022)


Nuitka

Nuitka is a Python compiler written in Python.  It's fully compatible with Python 2.6, 2.7, 3.4, 3.5, 3.6, 3.7, 3.8, 3.9, 3.10, and 3.11. You feed it your Python app, it does a lot of clever things, and spits out an executable or extension module.


Yatai

Model Deployment at Scale on Kubernetes 🦄


PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.


angel

A Flexible and Powerful Parameter Server for large-scale machine learning


budgetml

Deploy a ML inference service on a budget in less than 10 lines of code.


An easy to use PyTorch to TensorRT converter


heterogeneity-aware-lowering-and-optimization


TNN

TNN: developed by Tencent Youtu Lab and Guangying Lab, a uniform deep learning inference framework for mobile、desktop and server. TNN is distinguished by several outstanding features, including its cross-platform capability, high performance, model compression and code pruning. Based on ncnn and Rapidnet, TNN further strengthens the support and performance optimization for mobile devices, and also draws on the advantages of good extensibility and high performance from existed open source efforts. TNN has been deployed in multiple Apps from Tencent, such as Mobile QQ, Weishi, Pitu, etc. Contributions are welcome to work in collaborative with us and make TNN a better framework.


graphsignal-python

Graphsignal Python tracer


ML-Model-CI

MLModelCI is a complete MLOps platform for managing, converting, profiling, and deploying MLaaS (Machine Learning-as-a-Service), bridging the gap between current ML training and serving systems.


GFocal

Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection, NeurIPS2020


A scalable inference server for models optimized with OpenVINO™


network-bending

Manipulating the inner representations of StyleGAN2


ml-gmpi

Official PyTorch implementation of GMPI (ECCV 2022, Oral Presentation)


transfiner

Mask Transfiner for High-Quality Instance Segmentation, CVPR 2022


pydastic

An Elasticsearch Python ORM based on Pydantic.


Diffusion-LM

Diffusion-LM


supmae

This is a offical PyTorch/GPU implementation of SupMAE.


OpenDelta

A plug-and-play library for parameter-efficient-tuning (Delta Tuning)


imageio

Python library for reading and writing image data


git-cliff

A highly customizable Changelog Generator that follows Conventional Commit specifications ⛰


fps

Fast Pluggable Server


optimizers

For optimization algorithm research and development.


Drawflow

Simple flow library 🖥🖱


promptsource

Toolkit for creating, sharing and using natural language prompts.


tensorly

TensorLy: Tensor Learning in Python.


apptainer

Apptainer: Application containers for Linux


adaptdl

Resource-adaptive cluster scheduler for deep learning training.


tiptop

:desktop_computer: Command-line system monitoring


awful-ai

😈Awful AI is a curated list to track current scary usages of AI - hoping to raise awareness


Transformer-in-Vision

Recent Transformer-based CV and related works.


marktext

📝A simple and elegant markdown editor, available for Linux, macOS and Windows.


nerf.jax

A minimal TPU compatible Jax implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis.


stabilizer

Stabilize and achieve excellent performance with transformers


HyperbolicImageSegmentation

Hyperbolic Image Segmentation, CVPR 2022


vizzu-lib

Library for animated data visualizations and data stories.


redo

Redo is the ultimate tool to create reusable functions from your history in an interactive way


vectorhub

Vector Hub - Library for easy discovery, and consumption of State-of-the-art models to turn data into vectors. (text2vec, image2vec, video2vec, graph2vec, bert, inception, etc)


PyTorch code for "VL-Adapter: Parameter-Efficient Transfer Learning for Vision-and-Language Tasks" (CVPR2022)


neuron-descriptions

Natural Language Descriptions of Deep Visual Features, ICLR 2022


surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.


clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]


dataset-spec

Describe the format of image/text datasets


fastformers

FastFormers - highly efficient transformer models for NLU


bert-solr-search

Search with BERT vectors in Solr, Elasticsearch, OpenSearch and GSI APU


numba

NumPy aware dynamic Python compiler using LLVM


A collection of pretrained models for StyleGAN3


FastGAN-pytorch

Official implementation of the paper "Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis" in ICLR 2021


minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.


DiscoDiffusion-Warp

DiscoDiffusion Warp


dream-bench

A tool for benchmarking image generation models.


flamingo-pytorch

Implementation of 🦩 Flamingo, state-of-the-art few-shot visual question answering attention net out of Deepmind, in Pytorch


metaseq

Repo for external large-scale work


markdoc

A powerful, flexible, Markdown-based authoring framework.


self-label

Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)


Implementation of the video diffusion model and training scheme presented in the paper, Flexible Diffusion Modeling of Long Videos, in Pytorch


editpro

Learning to Model Editing Processes


A lightweight pythonic way to check if a package's dependencies are out of date


hyperlearn

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.


OpenPrompt

An Open-Source Framework for Prompt-Learning.


[ECCV 2022] Compositional Generation using Diffusion Models


jax-models

Unofficial JAX implementations of deep learning research papers


rax

Rax is a Learning-to-Rank library written in JAX.


semantic-segmentation

SOTA Semantic Segmentation Models in PyTorch


PromptPapers

Must-read papers on prompt-based tuning for pre-trained language models.


floral-diffusion

Floral Diffusion is a custom diffusion model trained by jags using a DD 5.6 version


bloxs

Build dashboards in Jupyter Notebook with numeric and chart boxes


CLIP-Caption-Reward

PyTorch code for "Fine-grained Image Captioning with CLIP Reward" (Findings of NAACL 2022)


bulk

A Simple Bulk Labelling Tool


quaterion-models

The collection of bulding blocks building fine-tunable metric learning models


Turn SymPy expressions into trainable JAX expressions.


p5 is a Python package based on the core ideas of Processing.


TurboTransformers

a fast and user-friendly runtime for transformer inference (Bert, Albert, GPT2, Decoders, etc) on CPU and GPU.


mmflow

OpenMMLab optical flow toolbox and benchmark


CogView2 for GPUs with 12/16/24GB vRAM


DeFlow

Official implementation of the paper DeFlow: Learning Complex Image Degradations from Unpaired Data with Conditional Flows


ipygany

3-D Scientific Visualization in the Jupyter Notebook


3D transformations for Python.


opengv

OpenGV is a collection of computer vision methods for solving geometric vision problems. It is hosted and maintained by the Mobile Perception Lab of ShanghaiTech.


3 dimensional spatial transformations


pyscript

Try PyScript: https://pyscript.com  Examples: https://tinyurl.com/pyscript-examples  Community: https://discord.gg/HxvBtukrg2


A framework for creating streamlit apps from markdown and python


[CVPR 2022] Incremental Transformer Structure Enhanced Image Inpainting with Masking Positional Encoding


awesome-self-supervised-learning

A curated list of awesome self-supervised methods


Code release for "Omni3D A Large Benchmark and Model for 3D Object Detection in the Wild"


Optimus

Optimus: the first large-scale pre-trained VAE language model


ghapi

A delightful and complete interface to GitHub's amazing API


cache

Cache dependencies and build outputs in GitHub Actions


sharpened-cosine-similarity

An alternative to convolution in neural networks


cascades

Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference, and more.


disco-diffusion-pbn

add paint-by-number functionality to disco diffusion


voxgraf

Official code release for VoxGRAF: Fast 3D-Aware Image Synthesis with Sparse Voxel Grids


Minimal standalone example of diffusion model


backend

Common source, scripts and utilities for creating Triton backends.


server

The Triton Inference Server provides an optimized cloud and edge inferencing solution.


The Triton backend for the ONNX Runtime.


Make-A-Scene

Pytorch implementation of Make-A-Scene: Scene-Based Text-to-Image Generation with Human Priors


laionide

checkpoints for glide finetuned on laion and other datasets. wip.


latent-diffusion-deepspeed

Finetune the 1.4B latent diffusion text2img-large checkpoint from CompVis using deepspeed. (work-in-progress)


Awesome-Video-Datasets

Video datasets


awesome-ebm

Collecting research materials on EBM/EBL (Energy Based Models, Energy Based Learning)


MAGIC

Language Models Can See: Plugging Visual Controls in Text Generation


laion-datasets

Description and pointers of laion datasets


IntraPaint

An interface for guiding AI image generation to make unique creative works.


Python optical flow visualization following Baker et al. (ICCV 2007) as used by the MPI-Sintel challenge


torchshow

Visualize PyTorch tensors with a single line of code.


tarp

Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.


CLIMS

[CVPR 2022] CLIMS: Cross Language Image Matching for Weakly Supervised Semantic Segmentation


CCAM

[CVPR 2022] C2AM: Contrastive learning of Class-agnostic Activation Map for Weakly Supervised Object Localization and Semantic Segmentation


Awesome-CLIP

Awesome list for research on CLIP (Contrastive Language-Image Pre-Training).


poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)


LeViT

LeViT a Vision Transformer in ConvNet's Clothing for Faster Inference


EfficientFormer

EfficientFormerV2 [ICCV 2023] & EfficientFormer [NeurIPs 2022]


mattermost

Mattermost is an open source platform for secure collaboration across the entire software development lifecycle..


artbench

Benchmarking Generative Models with Artworks


dino-vit-features

Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".


LiftedGAN

(CVPR 2021) Lifting 2D StyleGAN for 3D-Aware Face Generation


notebooks

Some notebooks for NLP


pypi-rename

Cookiecutter template for creating renamed PyPI packages


showyourwork

A workflow for reproducible and open scientific articles


jpu

JAX + Units


emcee

The Python ensemble sampling toolkit for affine-invariant MCMC


Apply black to ipynb files


maua

Deep learning toolkit for image, video, and audio synthesis


parsimonious

The fastest pure-Python PEG parser I can muster


ply

Python Lex-Yacc


lark

Lark is a parsing toolkit for Python, built with a focus on ergonomics, performance and modularity.


pregex

PRegEx - Programmable Regular Expressions


clip-retrieval

Easily compute clip embeddings and build a clip retrieval system with them


lm-evaluation-harness

A framework for few-shot evaluation of language models.


Bayesian-Julia

Bayesian Statistics using Julia and Turing

Creative Commons Attribution Share Alike 4.0 International


AdelaiDepth

This repo contains the projects: 'Virtual Normal', 'DiverseDepth', and '3D Scene Shape'. They aim to solve the monocular depth estimation, 3D scene reconstruction from single image problems.


dot

The Deepfake Offensive Toolkit


gdrcopy

A fast GPU memory copy library based on NVIDIA GPUDirect RDMA technology


tianshou

An elegant PyTorch deep reinforcement learning library.


river

🌊 Online machine learning in Python


ivy

The Unified AI Framework


simupy

A framework for modeling and simulating dynamical systems


gym

Fully differentiable RL environments, written in Ivy.


alf

Agent Learning Framework https://alf.readthedocs.io


GamiNet

GAMI-Net: Generalized Additive Models with Structured Interactions


wandb

🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.


streamlit

Streamlit — A faster way to build and share data apps.


bqplot

Plotting library for IPython/Jupyter notebooks


mpire

A Python package for easy multiprocessing, but faster than multiprocessing


pylint

It's not just a linter that annoys you!


leaps

Code for Learning to Synthesize Programs as Interpretable and Generalizable Policies in NeurIPS 2021


obs-studio

OBS Studio - Free and open source software for live streaming and screen recording


gymnax

RL Environments in JAX  🌍


PyFlow

An open-source tool for visual and modular block programming in python


Implementation of stylegan2 using deepspeed


snoop

A powerful set of Python debugging tools, based on PySnooper


pydeps

Python Module Dependency graphs


wtfpython

What the f*ck Python? 😱

Do What The F*ck You Want To Public License


textual

The lean application framework for Python.  Build sophisticated user interfaces with a simple Python API. Run your apps in the terminal and a web browser.


pytest-codeblocks

:page_facing_up: Test code blocks in your READMEs


patchy

:anchor: Patch the inner source of python functions at runtime.


plum

Multiple dispatch in Python


pyautogui

A cross-platform GUI automation Python module for human beings. Used to programmatically control the mouse & keyboard.


awesome-python

An opinionated list of awesome Python frameworks, libraries, software and resources.


ScreenToGif

🎬 ScreenToGif allows you to record a selected area of your screen, edit and save it as a gif or video.


ffmpeg-python

Python bindings for FFmpeg - with complex filtering support


vidgear

A High-performance cross-platform Video Processing Python framework powerpacked with unique trailblazing features :fire:


Natron

Open-source video compositing software. Node-graph based. Similar in functionalities to Adobe After Effects and Nuke by The Foundry.


ideas

Easy creation of custom import hooks to experiment on alternatives to Python's syntax; see https://aroberge.github.io/ideas/docs/html/


nushell

A new type of shell


opentelemetry-python-contrib

OpenTelemetry instrumentation for Python modules


IFM

Code for paper "Can contrastive learning avoid shortcut solutions?" NeurIPS 2021.


Robust Contrastive Learning Using Negative Samples with Diminished Semantics (NeurIPS 2021)


A comprehensive list of awesome contrastive self-supervised learning papers.


cruft

Allows you to maintain all the necessary cruft for packaging and building projects separate from the code you intentionally write. Built on-top of, and fully compatible with, CookieCutter.


neural-magic-eye

Neural Magic Eye: Learning to See and Understand the Scene Behind an Autostereogram, arXiv:2012.15692.


NeRF-SR

[ACM MM 2022] NeRF-SR: High-Quality Neural Radiance Fields using Supersampling


PREF

PREF: Phasorial Embedding Fields for Compact Neural Representations


A-ViT

Official PyTorch implementation of A-ViT: Adaptive Tokens for Efficient Vision Transformer (CVPR 2022)


vpt

❄🔥 Visual Prompt Tuning [ECCV 2022] https://arxiv.org/abs/2203.12119


jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/


quarto-cli

Open-source scientific and technical publishing system built on Pandoc.


python-template

Simple python template


The code for ECCV22 paper "Fast-Vid2Vid: Spatial-Temporal Compression for Video-to-Video Synthesis"


packr

Python package packer


Dynamic Neural Radiance Fields for Monocular 4D Facial Avater Reconstruction


jupyter-scatter

Interactive 2D scatter plot widget for Jupyter Lab and Notebook. Scales to millions of points!


EmbeddingRecycling

Embedding Recycling for Language models


dfdx

Deep learning in Rust, with shape checked tensors and neural networks


DallEval

DALL-Eval: Probing the Reasoning Skills and Social Biases of Text-to-Image Generation Models (ICCV 2023)


orbit

Experimental spaced repetition platform for exploring ideas in memory augmentation and programmable attention


pixel

Research code for pixel-based encoders of language (PIXEL)


textX

Domain-Specific Languages and parsers in Python made easy http://textx.github.io/textX/


TransGAN

[NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang


flash-attention

Fast and memory-efficient exact attention


vit-vqgan

JAX implementation ViT-VQGAN


ion

Real-Distributed  RTC System by pure Go and Flutter


bigbluebutton

Complete open source web conferencing system.


deskreen

Deskreen turns any device with a web browser into a secondary screen for your computer. ⭐ Star to support our work!


jibri

Jitsi BRoadcasting Infrastructure


jitsi-meet

Jitsi Meet - Secure, Simple and Scalable Video Conferences that you use as a standalone app or embed in your web application.


dendrite

Dendrite is a second-generation Matrix homeserver written in Go!


twirpy

Twirp's python implementation


twirp

A simple RPC framework with protobuf service definitions


livekit-server-sdk-python

LiveKit Server SDK for Python


livekit

End-to-end stack for WebRTC. SFU media server and SDKs.


sipvicious

SIPVicious OSS is a VoIP security testing toolset. It helps security teams, QA and developers test SIP-based VoIP systems and applications. This toolset is useful in simulating VoIP hacking attacks against PBX systems especially through identification, scanning, extension enumeration and password cracking.


homer

HOMER - 100% Open-Source SIP, VoIP, RTC Packet Capture & Monitoring


flutter-webrtc

WebRTC plugin for Flutter Mobile/Desktop/Web


Rocket.Chat

The communications platform that puts data protection first.


aiortc

WebRTC and ORTC implementation for Python using asyncio


peer-calls

Group peer to peer video calls for everyone written in Go and TypeScript


webrtc

Pure Go implementation of the WebRTC API


example-webrtc-applications

Examples of WebRTC applications that are large, or use 3rd party libraries


janus-gateway

Janus WebRTC Server


glow

Compiler for Neural Network hardware accelerators


visidata

A terminal spreadsheet multitool for discovering and arranging data


unzip-http

Extract individual files from .zip files over http without downloading the entire archive.


rlkit

Collection of reinforcement learning algorithms


braceexpand

Bash-style brace expansion for Python


Papers and such on 3D data compression and streaming


nglod

Neural Geometric Level of Detail: Real-time Rendering with Implicit 3D Shapes (CVPR 2021 Oral)


jupyterplot

Create real-time plots in Jupyter Notebooks.


Rust bindings for the Python interpreter


GPU-Puzzles

Solve puzzles. Learn CUDA.


This is the repo for my experiments with StyleGAN2. There are many like it, but this one is mine. Contains code for the paper Audio-reactive Latent Interpolations with StyleGAN.


singularity-tutorial

Tutorial for using Singularity containers


slurm-for-ml

A Machine Learning workflow for Slurm.


MeasureTheory.jl

"Distributions" that might not add to one.


Tilde.jl

WIP successor to Soss.jl


Metatheory.jl

General purpose algebraic metaprogramming and symbolic computation library for the Julia programming language: E-Graphs & equality saturation, term rewriting and more.


Soss.jl

Probabilistic programming via source rewriting


enhancing-transformers

An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch


torchestrator

Spin up Tor containers and then proxy HTTP requests via these Tor instances


upgini

Data search & enrichment library for Machine Learning → Easily find and add relevant features to your ML & AI pipeline from hundreds of public and premium external data sources, including open & commercial LLMs


GON

Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder


maggy

Distribution transparent Machine Learning experiments on Apache Spark


Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2


deepC

vendor independent TinyML deep learning library, compiler and inference framework microcomputers and micro-controllers


onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator


docker-freepbx

Dockerized FreePBX 15 w/Asterisk 17, Seperate MySQL Database support, and Data Persistence and UCP


multiplicative-filter-networks

Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.


ckconv

Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/2102.02611


cc-pyspark

Process Common Crawl data with Python and Spark


pytorch-grad-cam

Advanced AI Explainability for computer vision.  Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.


TimeSformer

The official pytorch implementation of our paper "Is Space-Time Attention All You Need for Video Understanding?"


Motionformer

Code + pre-trained models for the paper Keeping Your Eye on the Ball Trajectory Attention in Video Transformers


A PyTorch implementation of VIOLET


awesome-rvos

Referring Video Object Segmentation / Multi-Object Tracking Repo


Code release for ActionFormer (ECCV 2022)


VideoMAE

[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training


ml-cvnets

CVNets: A library for training computer vision networks


REVECA

Generic Event Boundary Captioning (GEBC) Challenge at LOVEU@CVPR 2022 - 3rd place (REVECA)


towhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.


st-nerf

PyTorch implementations for our SIGGRAPH 2021 paper: Editable Free-viewpoint Video Using a Layered Neural Representation.


MMVID

[CVPR 2022] Show Me What and Tell Me How: Video Synthesis via Multimodal Conditioning


mmrazor

OpenMMLab Model Compression Toolbox and Benchmark.


TecoGAN

This repo contains source code and materials for the TEmporally COherent GAN SIGGRAPH project.


deep-weight-prior

The Deep Weight Prior, ICLR 2019


eagle

Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models


geomloss

Geometric loss functions between point clouds, images and volumes


HGNN

Hypergraph Neural Networks (AAAI 2019)


math

The Stan Math Library is a C++ template library for automatic differentiation of any order using forward, reverse, and mixed modes.  It includes a range of built-in functions for probabilistic modeling, linear algebra, and equation solving.


A comprehensive collection of recent papers on graph deep learning


optnet

OptNet: Differentiable Optimization as a Layer in Neural Networks


pytorch-msssim

PyTorch differentiable Multi-Scale Structural Similarity (MS-SSIM) loss


optimesh

:spider_web: Mesh optimization, mesh smoothing.


DEODR

A differentiable 3D renderer with Pytorch, Tensorflow and Matlab interfaces


DISN

(latest updates and bug fixed) DISN: Deep Implicit Surface Network for High-quality Single-view 3D Reconstruction


serve

Serve, optimize and scale PyTorch models in production


pytorch-fid

Compute FID scores with PyTorch.


nanogui

Minimalistic C++/Python GUI library for OpenGL, GLES2/3, Metal, and WebAssembly/WebGL


nflows

Normalizing flows in PyTorch


manifold-flow

Manifold-learning flows (ℳ-flows)


meshoptimizer

Mesh optimization library that makes meshes smaller and faster to render


disentanglement-pytorch

Disentanglement library for PyTorch


nerfmeshes

A pipeline to reconstruct 3d meshes based on Neural Radiance Fields


mindspore

MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.


higan

[IJCV 2020] Semantic Hierarchy Emerges in Deep Generative Representations for Scene Synthesis


v

Simple, fast, safe, compiled language for developing maintainable software. Compiles itself in <1s with zero library dependencies. Supports automatic C => V translation. https://vlang.io


triton

Development repository for the Triton language and compiler


geonerf

Fork of https://github.com/krrish94/nerf-pytorch


pointnerf

Point-NeRF: Point-based Neural Radiance Fields


Lightweight MCMC sampling for PyTorch Models aka My Corona Project


convnet-burden

Memory consumption and FLOP count estimates for convnets


spytorch

Tutorial for surrogate gradient learning in spiking neural networks


SynapseML

Simple and Distributed Machine Learning


flaxOptimizers

A collection of optimizers, some arcane others well known, for Flax.


NeuRay

[CVPR2022] Neural Rays for Occlusion-aware Image-based Rendering


athens

ATHENS IS NO LONGER BEING ACTIVELY MAINTAINED. Athens is an open-source, collaborative knowledge graph.


simulacrabot

Discord AI Generation Bot to collect an aesthetic rating dataset


flexconv

Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.net/forum?id=3jooF27-0Wy


ccnn

Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs/2301.10540.


equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/


diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/


picologging

An optimized logging library for Python


EdgeNeXt

[CADL'22, ECCVW] Official repository of paper titled "EdgeNeXt: Efficiently Amalgamated CNN-Transformer Architecture for Mobile Vision Applications".


RAT-GAN

A conditional GAN for text-to-image


clip-gen

CLIP-GEN: Language-Free Training of a Text-to-Image Generator with CLIP


DF-GAN

A Simple and Effective Baseline for Text-to-Image Synthesis (CVPR2022 oral)


Awesome-Text-to-Image

(ෆ`꒳´ෆ) A Survey on Text-to-Image Generation/Synthesis.


edgedb

A graph-relational database with declarative schema, built-in migration system, and a next-generation query language


Tooll 3 is an open source software to create realtime motion graphics.


TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT


Instant-ngp in pytorch+cuda trained with pytorch-lightning (high quality with high speed, with only few lines of legible code)


deeplearning-models

A collection of various deep learning architectures, models, and tips


DeepFaceLive

Real-time face swap for PC streaming or video calls


searx

Privacy-respecting metasearch engine


Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)


you-get

:arrow_double_down: Dumb downloader that scrapes the web


pygtrie

Python library implementing a trie data structure.


Transformers-Tutorials

This repository contains demos I made with the Transformers library by HuggingFace.


Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models


SDEdit

PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations


Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch


PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)


improved-diffusion

Release for Improved Denoising Diffusion Probabilistic Models


v-diffusion-pytorch

v objective diffusion inference code for PyTorch.


deis

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


blended-latent-diffusion

Official implementation for "Blended Latent Diffusion" [SIGGRAPH 2023]


PyTorch reimplementation of Diffusion Models


Pipe

A Python library to use infix notation in Python


all-contributors

✨ Recognize all contributors, not just the ones who push code ✨


SDE

Example codes for the book Applied Stochastic Differential Equations


vnn

Vector Neurons: A General Framework for SO(3)-Equivariant Networks


🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥


volrend

PlenOctree Volume Rendering (supports CUDA & fragment shader backends)


HFS

HF-NeuS: Improved Surface Reconstruction Using High-Frequency Details (NeurIPS 2022)


Differender

Taichi-based Differentiable DVR Renderer


httpx

A next generation HTTP client for Python. 🦋


scenic

Scenic: A Jax Library for Computer Vision Research and Beyond


neuronika

Tensors and dynamic neural networks in pure Rust.


blackboard

The simple productivity logger.


doubtlab

Doubt your data, find bad labels.


remix

Build Better Websites. Create modern, resilient user experiences with web fundamentals.


taichi

Productive & portable high-performance programming in Python.


differentiable-programming-handbook

Differentiable implementation of common computer science algorithms


modern-srwm

Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)


owl

Owl - OCaml Scientific Computing @ http://ocaml.xyz


pulumi

Pulumi - Infrastructure as Code in any programming language. Build infrastructure intuitively on any cloud using familiar languages 🚀


evaluate

🤗 Evaluate: A library for easily evaluating machine learning models and datasets.


pytest-benchmark

py.test fixture for benchmarking code


simpletransformers

Transformers for Information Retrieval, Text Classification, NER, QA, Language Modelling, Language Generation, T5, Multi-Modal, and Conversational AI


Artistic Radiance Fields


deep-image-diffusion-prior

Inverts CLIP text embeds to image embeds and visualizes with deep-image-prior.


fast-neural-style

Feedforward style transfer


pytorch-generative

Easy generative modeling in PyTorch.


FENeRF

[CVPR 2022] FENeRF: Face Editing in Neural Radiance Fields


paxml

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.


swapping-autoencoder-pytorch

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


contrastive-unpaired-translation

Contrastive unpaired image-to-image translation, faster and lighter training than cyclegan (ECCV 2020, in PyTorch)


min-dalle

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch


torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.


awesome-normalizing-flows

Awesome resources on normalizing flows.


DST

Deformable Style Transfer (ECCV 2020)


A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.


impersonator

PyTorch implementation of our ICCV 2019 paper:  Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis


Pretrained Dalle2 from laion


discoart

🪩 Create Disco Diffusion artworks in one line


Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.


PlayableVideoGeneration

Official Pytorch implementation of "Playable Video Generation", CVPR 2022 (Oral)


The repository for CVPR 2022 Paper "Neural 3D Video Synthesis"


xfields

Source code for xfields (SIGGRAPH Asia 2020)


neural-light-fields

This repository contains the code for Learning Neural Light Fields with Ray-Space Embedding Networks.


generative-inverse-heat-dissipation

Code release for the paper Generative Modeling With Inverse Heat Dissipation


We estimate dense, flicker-free, geometrically consistent depth from monocular video, for example hand-held cell phone video.


Animate image in real time using First Order Motion Model for Image Animation


MiVOS

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!


DeepHomography

Content-Aware Unsupervised Deep Homography Estimation


OmniPhotos

Source code for "OmniPhotos: Casual 360° VR Photography"


PSFRGAN

PyTorch codes for "Progressive Semantic-Aware Style Transformation for Blind Face Restoration", CVPR2021


sceneformer

Generate indoor scenes with Transformers


SimDeblur

Simple framework for  image and video deblurring, implemented by PyTorch


prism

High Resolution Style Transfer in PyTorch with Color Control and Mixed Precision :art:


FLAVR

Code for FLAVR: A fast and efficient frame interpolation technique.


katna

Tool for automating common video key-frame extraction, video compression and Image Auto-crop/Image-resize tasks


flowframes

Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)


BasicSR

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also support StyleGAN2, DFDNet.


torch-dreams

Flexible Feature visualization on PyTorch, for research and art :mag_right: :computer: :brain: :art:


QueryInst

[ICCV 2021] Instances as Queries


[CVPR'21] DeepSurfels: Learning Online Appearance Fusion https://onlinereconstruction.github.io/DeepSurfels


SOLQ

"SOLQ: Segmenting Objects by Learning Queries", SOLQ is an end-to-end instance segmentation framework with Transformer.


webvid

Large-scale text-video dataset. 10 million captioned short videos.


frozen-in-time

Frozen in Time: A Joint Video and Image Encoder for End-to-End Retrieval [ICCV'21]


optimum-transformers

Accelerated NLP pipelines for fast inference on CPU and GPU. Built with Transformers, Optimum and ONNX Runtime.


hivemind

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.


pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision


DL course co-developed by YSDA, HSE and Skoltech


julia

The Julia Programming Language


electra

ELECTRA: Pre-training Text Encoders as Discriminators Rather Than Generators


gym

A toolkit for developing and comparing reinforcement learning algorithms.


colabgymrender

Render OpenAI Gym environments in Google Colaboratory


stanford-corenlp

Python wrapper for Stanford CoreNLP.


dask-cuda

Utilities for Dask and CUDA interactions


PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.


CLAP

Contrastive Language-Audio Pretraining


CLIP-ONNX

It is a simple library to speed up CLIP inference up to 3x (K80 GPU)


New-View-Synthesis

Collecting papers about new view synthesis


srf

Stereo Radiance Fields (SRF)  - Codebase


HairCLIP

[CVPR 2022] HairCLIP: Design Your Hair by Text and Reference Image


🚀 PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion Models(v-diffusion)"


ORViT

"Object-Region Video Transformers”, Herzig et al., CVPR 2022


videowalk

Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)


articulated-animation

Code for Motion Representations for Articulated Animation paper


warp-drive

Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)


CodeGen

CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.


OmniXAI

OmniXAI: A Library for eXplainable AI


MUST

PyTorch code for MUST


A Generative Language Model for Few-shot Aspect-Based Sentiment Analysis


ALPRO

Align and Prompt: Video-and-Language Pre-training with Entity Prompts


SRMA

Contrastive Learning with Model Augmentation


MoPro

MoPro: Webly Supervised Learning


progen

Official release of the ProGen models


NeuralWarp

Code release of paper Improving neural implicit surfaces geometry with patch warping


robot-inventor-tools

Tools to experiment with the LEGO® MINDSTORMS® Robot Inventor hub


mmappickle

Python 3 library to store memory mappable objects into pickle-compatible files


HyperInverter

HyperInverter: Improving StyleGAN Inversion via Hypernetwork (CVPR 2022)


Thin-Plate-Spline-Motion-Model

[CVPR 2022] Thin-Plate Spline Motion Model for Image Animation.


EGVSR

Efficient & Generic Video Super-Resolution


Code for "Learning Object-Compositional Neural Radiance Field for Editable Scene Rendering", ICCV 2021


pysot

SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.


ddsp

DDSP: Differentiable Digital Signal Processing


CS-DisMo

[ICCVW 2021] Rethinking Content and Style: Exploring Bias for Unsupervised Disentanglement


EDVR

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.


DenseDepth

High Quality Monocular Depth Estimation via Transfer Learning


DALLE-models

Here is a collection of checkpoints for DALLE-pytorch models, from where you can keep on training or start generating images.


image-color-replacement

Image color replacement using numerical optimization


StyleGAN2 fork with scripts and convenience modifications for creative media synthesis


Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation


MotionCLIP

Official Pytorch implementation of the paper "MotionCLIP: Exposing Human Motion Generation to CLIP Space"


super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.


video-diffusion-pytorch

Implementation of Video Diffusion Models, Jonathan Ho's new paper extending DDPMs to Video Generation - in Pytorch


JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization


VSGAN

PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth


pyxelate

Python class that generates pixel art from images


Improved StyleGAN Embedding: Where are the Good Latents?


ners

Code for "NeRS: Neural Reflectance Surfaces for Sparse-View 3D Reconstruction in the Wild," in NeurIPS 2021


LowRankGAN

[NeurIPS 2021] Low-Rank Subspaces in GANs


CLIPstyler

Official Pytorch implementation of "CLIPstyler:Image Style Transfer with a Single Text Condition" (CVPR 2022)


hyperstyle

Official Implementation for "HyperStyle: StyleGAN Inversion with HyperNetworks for Real Image Editing" (CVPR 2022) https://arxiv.org/abs/2111.15666


ESRGAN (Enhanced Super Resolution GAN) using two 2x2 (kernel size) conv2d layers instead of a traditional single 3x3 conv2d layer in its conv block.


Pyramid-Attention-Networks

[IJCV] Pyramid Attention Networks for Image Restoration: new SOTA results on multiple image restoration tasks: denoising, demosaicing, compression artifact reduction, super-resolution


The idea of this list is to collect shared data and algorithms around 3D Morphable Models. You are invited to contribute to this list by adding a pull request. The original list arised from the Dagstuhl seminar on 3D Morphable Models https://www.dagstuhl.de/19102 in March 2019.


VideoAutoencoder

Video Autoencoder: self-supervised disentanglement of 3D structure and motion (ICCV 2021). Website: https://zlai0.github.io/VideoAutoencoder/


AIM

[IJCAI'21] Deep Automatic Natural Image Matting


cyclic-cosine-decay

Pytorch cyclic cosine decay learning rate scheduler


Source code of "A structured dictionary perspective on implicit neural representations"


DSNeRF

Code release for DS-NeRF (Depth-supervised Neural Radiance Fields)


The codebase for our paper "Generative Occupancy Fields for 3D Surface-Aware Image Synthesis" (NeurIPS 2021)


DiscordChatExporter

Exports Discord chat logs to a file


get-started-with-JAX

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.


neural-dream

PyTorch implementation of DeepDream algorithm


AI-Art

PyTorch (and PyTorch Lightning) implementation of Neural Style Transfer, Pix2Pix, CycleGAN, and Deep Dream!


autocrop

Automatically crop and rotate scanned images using OpenCV


ldm-finetune

Home of `erlich` and `ongo`. Finetune latent-diffusion/glid-3-xl text2image on your own data.


Game-Console-Dev-Guide

Game Console Dev Guide. Learn to develop games for Xbox Series X|S, PlayStation 5, Nintendo Switch, Steam Deck, and Apple Silicon. Developing with Unreal Engine 5, Unity, Godot Engine, and Blender.


Seamless operability between C++11 and Python


Node based visual scripting system designed for motion graphics in Blender.


awesome-cg-vfx-pipeline

List of open-source technologies that help in the process of building a pipeline for CG and VFX productions


monkey-net

Animating Arbitrary Objects via Deep Motion Transfer


open-optimizers

Repository for open research on optimizers.


playtorch

PlayTorch is a framework for rapidly creating mobile AI experiences.


rtx-voice-script

A python script that takes an input MP3/FLAC and outputs an acapella/background noise stripped WAV using the power of NVIDIA's RTX Voice


awesome-neural-rendering

Resources of Neural Rendering


ImgAlign

For auto aligning, cropping, and scaling HR and LR images for training image based neural networks


GCVit

[ICML 2023] Official PyTorch implementation of Global Context Vision Transformers


ConvChain

Bitmap generation from a single example with convolutions and MCMC


CogVideo

Text-to-video generation. The repo for ICLR2023 paper "CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers"


GIRAFFEHD

Official Github repository for the CVPR 2022 paper "GIRAFFE HD: A High-Resolution 3D-aware Generative Model"


WaveFunctionCollapseEditor

Simple GUI for Bitmap & tilemap generation from a single example with the help of ideas from quantum mechanics.


A curated list of awesome 3d generation papers


NeuralNeighborStyleTransfer

Optimization based style transfer


Official code for "Towards An End-to-End Framework for Flow-Guided Video Inpainting" (CVPR2022)


diffae

Official implementation of Diffusion Autoencoders


AvatarCLIP

[SIGGRAPH 2022 Journal Track] AvatarCLIP: Zero-Shot Text-Driven Generation and Animation of 3D Avatars


GLIP

Grounded Language-Image Pre-training


Artistic Radiance Fields


volumegan

CVPR 2022 VolumeGAN - 3D-aware Image Synthesis via Learning Structural and Textural Representations


official code repo for paper "CogView2: Faster and Better Text-to-Image Generation via Hierarchical Transformers"


[CVPR 2022] GAN inversion and editing with spatially-adaptive multiple latent layers


nvdiffrec

Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".


x-unet

Implementation of a U-net complete with efficient attention as well as the latest research findings


DenseMTL

Cross-task Attention Mechanism for Dense Multi-task Learning (WACV 2023)


pollinations

Generate Art


epigraf

[NeurIPS 2022] Official pytorch implementation of EpiGRAF


k-diffusion

Karras et al. (2022) diffusion models for PyTorch


embedding-reader

Efficiently read embedding in streaming from any filesystem


ema-pytorch

A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model


Tiny PyTorch library for maintaining a moving average of a collection of parameters.


nos

Module to Automatically maximize the utilization of GPU resources in a Kubernetes cluster through real-time dynamic partitioning and elastic quotas - Effortless optimization at its finest!


dnn.cool

A framework for multi-task learning, where you may precondition tasks and compose them into bigger tasks. Conditional objectives and per-task evaluations and interpretations.


diffusers

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


Efficient distortion loss with O(n) realization.


hyperlib

Library that contains implementations of machine learning components in the hyperbolic space


mmap.ninja

Memory mapped numpy arrays of varying shapes


CLIPNeRF

CLIP-NeRF: Text-and-Image Driven Manipulation of Neural Radiance Fields


CLIP-Mesh-SMPLX

CLIP + Mesh + SMPL-X


SwissArmyTransformer

SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.


DefTet

Learning Deformable Tetrahedral Meshes for 3D Reconstruction (NeurIPS 2020)


Multicore-TSNE

Parallel t-SNE implementation with Python and Torch wrappers.


TMNet

The official pytorch implemention of the CVPR paper "Temporal Modulation Network for Controllable Space-Time Video Super-Resolution".


liif

Learning Continuous Image Representation with Local Implicit Image Function, in CVPR 2021 (Oral)


image-similarity-measures

:chart_with_upwards_trend: Implementation of eight evaluation metrics to access the similarity between two images. The eight metrics are as follows: RMSE, PSNR, SSIM, ISSM, FSIM, SRE, SAM, and UIQ.


Compact-Transformers

Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)


SinNeRF

[ECCV 2022] "SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Humphrey Shi, Zhangyang Wang


Fast and Accurate One-Stage Space-Time Video Super-Resolution (accepted in CVPR 2020)


[CVPR 2022] VideoINR: Learning Video Implicit Neural Representation for Continuous Space-Time Super-Resolution


Neighborhood-Attention-Transformer

[CVPR 2023] Neighborhood Attention Transformer and [arXiv] Dilated Neighborhood Attention Transformer repository.


mobilePydnet

Pydnet on mobile devices


LIVE-Layerwise-Image-Vectorization

[CVPR 2022 Oral] Towards Layer-wise Image Vectorization


NOAH

Searching prompt modules for parameter-efficient transfer learning.


StyleSpace

StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation


jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.


Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.


pytti-ebsynth

frame interpolation for CLIP guided videos


pint

Operate and manipulate physical quantities in Python


jax-ngp

JAX implementation of https://github.com/NVlabs/instant-ngp


Multilingual-CLIP

OpenAI CLIP text encoders for multiple languages!


bagua

Bagua Speeds up PyTorch


nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference


openvino

OpenVINO™ is an open-source toolkit for optimizing and deploying AI inference


voxblox

A library for flexible voxel-based mapping, mainly focusing on truncated and Euclidean signed distance fields.


TensoRF

[ECCV 2022] Tensorial Radiance Fields, a novel approach to model and reconstruct radiance fields


CLIP

Contrastive Language-Image Pretraining


ASSET

ASSET: Autoregressive Semantic Scene Editing with Transformers at High Resolutions (SIGGRAPH 2022 - Journal Track)


natsort

Simple yet flexible natural sorting in Python.


majesty-diffusion

Majesty Diffusion by @Dango233(@Dango233max) and @apolinario (@multimodalart)


dropit

DropIT: Dropping Intermediate Tensors for Memory-Efficient DNN Training (ICLR'23)


lazy

Lazy, a tool for running things in idle time


imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch


boxflow

Visual dataflow programming between parameterized objects


maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.


E(2)-Equivariant CNNs Library for Pytorch


theseus

A library for differentiable nonlinear optimization


omnivore

Omnivore: A Single Model for Many Visual Modalities


pynvml

Provide Python access to the NVML library for GPU diagnostics


Online Panoptic 3D Reconstruction


DeCLUTR

The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!


Neural-IMage-Assessment

A PyTorch Implementation of Neural IMage Assessment


repository-overview

This repository will give a quick overview of all projects and repositories from LAION.


alpa

Training and serving large-scale neural networks with auto parallelization.


notebooks

All of my Colab notebook


SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)


dalle-flow

🌊 A Human-in-the-Loop workflow for creating HD images from text


deep-image-prior

Image restoration with neural networks but without learning.


deep-image-prior-hqskipnet

A modification of Daniel Russell's notebook merged with Katherine Crowson's hq-skip-net changes


magic-wormhole

get things from one computer to another, safely


unleashing-transformers

Code for the ECCV 2022 paper "Unleashing Transformers"


STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences


subspace-diffusion

Improved diffusion generative models with subspaces


oneFace

Generating interfaces(CLI, Qt GUI, Dash web app) from a Python function or a command program.


pytest-rich

pytest + rich integration (proof of concept)


ColossalAI

Making large AI models cheaper, faster and more accessible


agentos

The Python Component System (PCS) is an API and CLI for building, running, and sharing Python code. AgentOS is a set of libraries built on top of PCS that make it easy to build, run, and share agents that use Reinforcement Learning.


rich-click

Format click help output nicely with rich.


organize

The file management automation tool.


clone-repo

A tool to easily clone repos


pdbr

pdb + Rich library


felicette

Satellite imagery for dummies.


norfair

Lightweight Python library for adding real-time multi-object tracking to any detector.


onnx-simplifier

Simplify your onnx model


gto

🏷 Git Tag Ops. Turn your Git repository into Artifact Registry or Model Registry.


memray

Memray is a memory profiler for Python


transformers

🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.


anime style transfer with pytorch


directclr

Code used in "Understanding Dimensional Collapse in Contrastive Self-supervised Learning" paper.


Processing

Processing sketches


PermutedAdaIN

Pytorch Implementation of PermutedAdaIN


dalle-mini

DALL·E Mini - Generate images from a text prompt


simple-diffusion-model

Pedagogical codebase for a simplified score-based generative model design, with training loop


DiffusionCLIP

[CVPR 2022] Official PyTorch Implementation for DiffusionCLIP: Text-guided Image Manipulation Using Diffusion Models


NeuralImageSuperResolution

Colabs for Neural Image Enhancement.


NextFace

A high-fidelity 3D face reconstruction library from monocular RGB image(s)


yt-dlp

A youtube-dl fork with additional features and fixes


Utils for streaming large files (S3, HDFS, gzip, bz2...)


nice-slam

[CVPR'22] NICE-SLAM: Neural Implicit Scalable Encoding for SLAM


StyleGANXL-CLIP

A notebook for text-based guided image generation using StyleGANXL and CLIP.


TASO

The Tensor Algebra SuperOptimizer for Deep Learning


my-colab-experiments

A repository for sharing ipynb's of my experiments with ML. Some notebooks are 'old' by now and might no longer work 'out of the box'.


probing-vits

Probing the representations of Vision Transformers.


demucs

Code for the paper Hybrid Spectrogram and Waveform Source Separation


torchdynamo

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.


mmf

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)


multimodal

TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.


CCL

PyTorch Implementation on Paper [CVPR2021]Distilling Audio-Visual Knowledge by Compositional Contrastive Learning


modality-distillation

Tensorflow code for the paper 'Modality Distillation with Multiple Stream Networks for Action Recognition', ECCV 2018


The pretrained models trained on Moments in Time Dataset


deq-flow

[CVPR 2022] Deep Equilibrium Optical Flow Estimation


PointCLIP

[CVPR 2022] PointCLIP: Point Cloud Understanding by CLIP


ASFormer

Official repo for BMVC2021 paper ASFormer: Transformer for action segmentation


Bridge-Prompt

[CVPR2022] Bridge-Prompt: Towards Ordinal Action Understanding in Instructional Videos


hy

A dialect of Lisp that's embedded in Python


Dassl.pytorch

A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.


PB-OVD

A pytorch Implementation of Open Vocabulary Object Detection with Pseudo Bounding-Box Labels


Tensor-Puzzles

Solve puzzles. Improve your pytorch.


IOPaint

Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.


KELIP

Official PyTorch implementation of "Large-scale Bilingual Language-Image Contrastive Learning" (ICLRW 2022)


difftastic

a structural diff that understands syntax 🟥🟩


esvit

EsViT: Efficient self-supervised Vision Transformers


CvT

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.


VQ-Diffusion

Official implementation of VQ-Diffusion


Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)


SimMIM

This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".


Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".


Olive

Olive is an easy-to-use hardware-aware model optimization tool that composes industry-leading techniques across model compression, optimization, and compilation.


Cream

This is a collection of our NAS and Vision Transformer work.


Pop-Out-Motion

Pop-Out Motion: 3D-Aware Image Deformation via Learning the Shape Laplacian (CVPR 2022)


decord

An efficient video loader for deep learning with smart shuffling that's super easy to digest


1.4B latent diffusion model fine tuning


pydash

The kitchen sink of Python utility libraries for doing "stuff" in a functional way. Based on the Lo-Dash Javascript library.


fnc

Functional programming in Python with generators and other utilities.


FinalRecon

All In One Web Recon


TATS

Official PyTorch implementation of TATS: A Long Video Generation Framework with Time-Agnostic VQGAN and Time-Sensitive Transformer (ECCV 2022)


Script and models for clustering LAION-400m CLIP embeddings.


denoising-diffusion-gan

Tackling the Generative Learning Trilemma with Denoising Diffusion GANs  https://arxiv.org/abs/2112.07804


maskgit

Official Jax Implementation of MaskGIT


poethepoet

A task runner that works well with poetry.


Implementation of popular SOTA self-supervised learning algorithms as Fastai Callbacks.


ActionCLIP

This is the official implement of paper "ActionCLIP: A New Paradigm for Action Recognition"


CoOp

Prompt Learning for Vision-Language Models (IJCV'22, CVPR'22)


wit

WIT (Wikipedia-based Image Text) Dataset is a large multimodal multilingual dataset comprising 37M+ image-text sets with 11M+ unique images across 100+ languages.


typeguard

Run-time type checker for Python


polars

Dataframes powered by a multithreaded, vectorized query engine, written in Rust


pigeon

🐦 Quickly annotate data from the comfort of your Jupyter notebook


nbval

A py.test plugin to validate Jupyter notebooks


pyinstrument

🚴 Call stack profiler for Python. Shows you why your code is slow!


parse

Parse strings using a specification based on the Python format() syntax.


cloob-latent-diffusion

CLOOB Conditioned Latent Diffusion training and inference code


fastcore

Python supercharged for the fastai library


BP-DIP-deblurring

Based on "BP-DIP: A Backprojection based Deep Image Prior"


CogView

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".


stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets


jaxton

100 exercises to learn JAX


BoostYourOwnDepth

Apply our monocular depth boosting to your own network!


imgbeddings

Python package to generate image embeddings with CLIP without PyTorch/TensorFlow


Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.


laion-prepro

Get hundred of million of image+url from the crawling at home dataset and preprocess them


nn-template

Generic template to bootstrap your PyTorch project.


GroupViT

Official PyTorch implementation of GroupViT: Semantic Segmentation Emerges from Text Supervision, CVPR 2022.


rq-vae-transformer

The official implementation of Autoregressive Image Generation using Residual Quantization (CVPR '22)


torch-ngp

A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.


Decoupled-Contrastive-Learning

This repository is an implementation for the loss function proposed in https://arxiv.org/pdf/2110.06848.pdf.


conerf

😮The official implementation of "CoNeRF: Controllable Neural Radiance Fields" 😮


DualStyleGAN

[CVPR 2022] Pastiche Master: Exemplar-Based High-Resolution Portrait Style Transfer


cloob-training

CLOOB training (JAX) and inference (JAX and PyTorch)


mmselfsup

OpenMMLab Self-Supervised Learning Toolbox and Benchmark


gudhi-devel

The GUDHI library is a generic open source C++ library, with a Python interface, for Topological Data Analysis (TDA) and Higher Dimensional Geometry Understanding.


keops

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows


toma

Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory


OneshotCLIP

Official Source code of "One-Shot Adaptation of GAN in Just One CLIP" IEEE Transactions on Pattern Anaylsis and Machine Intelligence (TPAMI)


LIA

[ICLR 22] Latent Image Animator: Learning to Animate Images via Latent Space Navigation


DAT

Repository of Vision Transformer with Deformable Attention (CVPR2022) and DAT++: Spatially Dynamic Vision Transformerwith Deformable Attention


PyTorch code for "Unifying Vision-and-Language Tasks via Text Generation" (ICML 2021)


emblaze

Interactive Jupyter notebook widget for visually comparing embedding spaces.


This repository contains the code for the CVPR 2020 paper "Differentiable Volumetric Rendering: Learning Implicit 3D Representations without 3D Supervision"


self-distilled-internet-photos

Creative Commons Attribution 4.0 International


StyleSwin

[CVPR 2022] StyleSwin: Transformer-based GAN for High-resolution Image Generation


magma

MAGMA - a GPT-style multimodal model that can understand any combination of images and language. NOTE: The freely available model from this repo is only a demo. For the latest multimodal and multilingual models from Aleph Alpha check out our website https://app.aleph-alpha.com


SIMAT

codebase for the SIMAT dataset and evaluation


pytorch-lamb

Implementation of  https://arxiv.org/abs/1904.00962


freemocap

Free Motion Capture for Everyone 💀✨


light-the-torch

Install PyTorch distributions with computation backend auto-detection


cutlass

CUDA Templates for Linear Algebra Subroutines


WeightWatcher

The WeightWatcher tool for predicting the accuracy of   Deep Neural Networks


Hyper-Convolution

Hyper-Convolution Networks for Biomedical Image Segmentation


HCSC

[CVPR 2022] PyTorch implementation of Hierarchical Contrastive Selective Coding (HCSC) (https://arxiv.org/abs/2202.00455)


Lafite

Code for paper LAFITE: Towards Language-Free Training for Text-to-Image Generation (CVPR 2022)


VRT

VRT: A Video Restoration Transformer (official repository)


SimSwap-colab

Colab notebooks to apply SimSwap to images, animated GIF, and videos.


Official PyTorch implementation of StyleGAN3


Style-AttnGAN

Improves Text to Image synthesis from AttnGAN by integrating the scale-specific control from StyleGAN; can optionally use GPT-2 as text encoder


StyleNeRF

This is the open source implementation of the ICLR2022 paper "StyleNeRF: A Style-based 3D-Aware Generator for High-resolution Image Synthesis"


nuwa-pytorch

Implementation of NÜWA, state of the art attention network for text to video synthesis, in Pytorch


HeroicGamesLauncher

A games launcher for GOG, Amazon and Epic Games for Linux, Windows and macOS.


neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.


tgn

TGN: Temporal Graph Networks


Training simple models to predict CLIP image embeddings from text embeddings, and vice versa.


wise-ft

Robust fine-tuning of zero-shot models


multimodalai

Multimodal AI related code


pytorch-blender

:sweat_drops: Seamless, distributed, real-time integration of Blender into PyTorch data pipelines


blended-diffusion

Official implementation for "Blended Diffusion for Text-driven Editing of Natural Images" [CVPR 2022]


spleeter

Deezer source separation library including pretrained models.


RWKV-LM

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.


StructuredDreaming

Repo for structured dreaming


artemis

Learning to ground explanations of affect for visual art.


CLD-SGM

Score-Based Generative Modeling with Critically-Damped Langevin Diffusion


StyleCLIP

Official Implementation for "StyleCLIP: Text-Driven Manipulation of StyleGAN Imagery" (ICCV 2021 Oral)


aphantasia

CLIP + FFT/DWT/RGB = text to image/video


frame-interpolation

FILM: Frame Interpolation for Large Motion, In ECCV 2022.


OFA

Official repository of OFA (ICML 2022). Paper: OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework


nebuly

The user analytics platform for LLMs


jukebox

Code for "Jukebox: A Generative Model for Music"


An open source implementation of CLIP.


hopfield-layers

Hopfield Networks is All You Need


SC-FEGAN

SC-FEGAN : Face Editing Generative Adversarial Network with User's Sketch and Color (ICCV2019)


mesh

Mesh TensorFlow: Model Parallelism Made Easier


HashNeRF-pytorch

Pure PyTorch Implementation of NVIDIA paper on Instant Training of Neural Graphics primitives: https://nvlabs.github.io/instant-ngp/


ryvencore-qt

Qt frontend for ryvencore - Python library for building visual node editors


Ryven

Flow-based visual scripting for Python


drawfbp

Tool for Creating and Exploring Flow-Based Programming Diagram Hierarchies


flowpipe

Very simple flow-based programming framework.


rete

JavaScript framework for visual programming


PyFlow

Visual scripting framework for python - https://wonderworks-software.github.io/PyFlow


nodeeditor

Qt Node Editor. Dataflow programming framework


Nodz

Nodz : Visualize your data - Free nodes based graph generator.


nodegraph-pyqt

A generic visual directed acyclic graph implementation powered by PyQT/PySide (and eventually NetworkX)


GANSketching

Sketch Your Own GAN: Customizing a GAN model with hand-drawn sketches.


latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models


VQGAN-CLIP

VQGAN+CLIP Colab Notebook with user-friendly interface.


tqdm-multiprocess

Using queues, tqdm-multiprocess supports multiple worker processes, each with multiple tqdm progress bars, displaying them cleanly through the main process. It offers similar functionality for python logging.


clipseg

This repository contains the code of the CVPR 2022 paper "Image Segmentation Using Text and Image Prompts".


mdmm

The Modified Differential Multiplier Method (MDMM) for PyTorch


python-prompt-toolkit

Library for building powerful interactive command line applications in Python


autowu

Official repository for Automated Learning Rate Scheduler for Large-Batch Training (8th ICML Workshop on AutoML)


sparse-detr

PyTorch Implementation of Sparse DETR


fast-autoaugment

Official Implementation of 'Fast AutoAugment' in PyTorch.


mindall-e

PyTorch implementation of a 1.3B text-to-image generation model trained on 14 million image-text pairs


FEAT

The code repository for "Few-Shot Learning via Embedding Adaptation with Set-to-Set Functions"


pytti-notebook

Start here


hyperbolic-image-embeddings

Supplementary code for the paper "Hyperbolic Image Embeddings".


vivit

[TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivatives & Newton steps


effect

effect isolation in Python, to facilitate more purely functional code


ocaml-effects-tutorial

Concurrent Programming with Effect Handlers


TTUR

Two time-scale update rule for training GANs


towncrier

Manage the release notes for your project.


mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.


rembg

Rembg is a tool to remove images background


responsible-ai-toolbox

Responsible AI Toolbox is a suite of tools providing model and data exploration and assessment user interfaces and libraries that enable a better understanding of AI systems. These interfaces and libraries empower developers and stakeholders of AI systems to develop and monitor AI more responsibly, and take better data-driven actions.


PNDM

The official implementation for Pseudo Numerical Methods for Diffusion Models on Manifolds (PNDM, PLMS | ICLR2022)


taming-transformers

Taming Transformers for High-Resolution Image Synthesis


Vision-Language-Transformer

[ICCV2021 & TPAMI2023] Vision-Language Transformer and Query Generation for Referring Segmentation


ReferFormer

[CVPR2022] Official Implementation of ReferFormer


Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"


OTTER

This code provides a PyTorch implementation for OTTER (Optimal Transport distillation for Efficient zero-shot Recognition), as described in the paper.


diffq

DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight or group of weights, in order to achieve a given trade-off between model size and accuracy.


denoiser

Real Time Speech Enhancement in the Waveform Domain (Interspeech 2020)We provide a PyTorch implementation of the paper Real Time Speech Enhancement in the Waveform Domain. In which, we present a causal speech enhancement model working on the raw waveform that runs in real-time on a laptop CPU. The proposed model is based on an encoder-decoder architecture with skip-connections. It is optimized on both time and frequency domains, using multiple loss functions. Empirical evidence shows that it is capable of removing various kinds of background noise including stationary and non-stationary noises, as well as room reverb. Additionally, we suggest a set of data augmentation techniques applied directly on the raw waveform which further improve model performance and its generalization abilities.


EGG

EGG: Emergence of lanGuage in Games


Detic

Code release for "Detecting Twenty-thousand Classes using Image-level Supervision".


Mephisto

A suite of tools for managing crowdsourcing tasks from the inception through to data packaging for research use.


MetaBalance

MetaBalance algorithm for multi-task learning


MTAdam

MTAdam: Automatic Balancing of Multiple Training Loss Terms


Conceptual 12M is a dataset containing (image-URL, caption) pairs collected for vision-and-language pre-training.


MoCoGAN-HD

[ICLR 2021 Spotlight] A Good Image Generator Is What You Need for High-Resolution Video Synthesis


VQGAN+CLIP with some additional tuning. For notebooks and the command line.


DECA

DECA: Detailed Expression Capture and Animation (SIGGRAPH 2021)


idinvert

[ECCV 2020] In-Domain GAN Inversion for Real Image Editing


Kervolution

Kervolution implementation using TF2.0


aesara

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.


pillow-simd

The friendly PIL fork


torch-fidelity

High-fidelity performance metrics for generative models in PyTorch


Video-Inference

Easy inference for video networks. Currently supports SOFVSR (traiNNer Version), RIFE, and TecoGAN-pytorch


ESRGAN

A modified version of the original ESRGAN test.py script with added features


chaiNNer

A node-based image processing GUI aimed at making chaining image processing tasks easy and customizable. Born as an AI upscaling application, chaiNNer has grown into an extremely flexible and powerful programmatic image processing application.


sacremoses

Python port of Moses tokenizer, truecaser and normalizer


AudioCLIP

Source code for models described in the paper "AudioCLIP: Extending CLIP to Image, Text and Audio" (https://arxiv.org/abs/2106.13043)


Trans-CycleGAN

A convolution-free, transformer-only version of the CycleGAN framework


3D mesh stylization driven by a text input in PyTorch


exa

A modern replacement for ‘ls’.


ALBEF

Code for ALBEF: a new vision-language pre-training method


BLIP

PyTorch code for BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generation


docusaurus

Easy to maintain open source documentation websites.


python-hunter

Hunter is a flexible code tracing toolkit.


cookiecutter-pylibrary

Enhanced cookiecutter template for Python libraries.


EssentialMC2 Video Understanding.


TAdaConv

[ICLR 2022] TAda! Temporally-Adaptive Convolutions for Video Understanding. This codebase provides solutions for video classification, video representation learning and temporal detection.


dense-ulearn-vos

Dense Unsupervised Learning for Video Segmentation (NeurIPS*2021)


ESAN

Equivariant Subgraph Aggregation Networks (ICLR 2022 Spotlight)


Class site for ITP’s 2022 Making Visual Art with GANs


pl-class-public

Public slides and assignments for CSCI-UA.04900: Special Topics in Programming Languages


SuperPoint

Efficient neural feature detector and descriptor


openfold

Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2


pooch

A friend to fetch your data files


pymunk

Pymunk is a easy-to-use pythonic 2d physics library that can be used whenever  you need 2d rigid body physics from Python


NN-SVG

Publication-ready NN-architecture schematics.


RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!


VIBE

Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"


DiffuseVAE

Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"


RePaint

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


netdata

The open-source observability platform everyone needs!


muzero-general

MuZero


PyTorch implementation of DQN, AC,  ACER, A2C, A3C, PG,  DDPG, TRPO, PPO, SAC, TD3 and ....


conky

Light-weight system monitor for X, Wayland, and other things, too


pydags

Simple, lightweight, extensible DAG framework for Python with a Kubeflow-like API


implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets


composer

Visual and code editor for Common Workflow Language


openFrameworks

openFrameworks is a community-developed cross platform toolkit for creative coding in C++.


synsin

View synthesis for the public.


softmax-splatting

an implementation of softmax splatting for differentiable forward warping using PyTorch


AR-Depth

Fast Depth Densification for Occlusion-Aware Augmented Reality


calibration-framework

The net:cal calibration framework is a Python 3 library for measuring and mitigating miscalibration of uncertainty estimates, e.g., by a neural network.


Laplace

Laplace approximations for Deep Learning.


laplace-redux

Laplace Redux -- Effortless Bayesian Deep Learning


MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.


CLIP-ViL

[ICLR 2022] code for "How Much Can CLIP Benefit Vision-and-Language Tasks?" https://arxiv.org/abs/2107.06383


SLIP

Code release for SLIP Self-supervision meets Language-Image Pre-training


SWAG

Official repository for "Revisiting Weakly Supervised Pre-Training of Visual Perception Models". https://arxiv.org/abs/2201.08371.


RETRO-pytorch

Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch


homelab

Modern self-hosting framework, fully automated from empty disk to operating services with a single command.


jugglemaster

A multi-platform siteswap juggling animator


noclip.website

A digital museum of video game levels


pytorch-caffe-models

The original weights of some Caffe models, ported to PyTorch.


ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)


Widget generation from a traitlets HasTraits model


ipychart

The power of Chart.js with Python


ipyflex

A WYSIWYG layout editor for Jupyter widgets


openvdb

OpenVDB - Sparse volume data structure and tools


tiny-cuda-nn

Lightning fast C++/CUDA neural network framework


nex-code

Code release for NeX: Real-time View Synthesis with Neural Basis Expansion


XNNPACK

High-efficiency floating-point neural network inference operators for mobile, server, and Web


combination of OpenAI GLIDE and Latent Diffusion


L-Verse

L-Verse: Bidirectional Generation Between Image and Text


instant-ngp

Instant neural graphics primitives: lightning fast NeRF and more


click

Python composable command line interface toolkit


tranquilizer

Put your functions to REST


hiraishin

A lightning rod for DL Models


everett

configuration library for python projects


clearml-web

ClearML - Web Application - Auto-Magical Suite of tools to streamline your ML workflow. Experiment Manager, ML-Ops and Data-Management


mlflow-elasticsearchstore

ElasticSearch implementation of MlFlow tracking store


mlflow-docker-compose

Deploy mlflow with docker-compose


minio

The Object Store for AI Data Infrastructure


mlflow-docker

Ready to run docker-compose configuration for ML Flow with Mysql and Minio S3


lightning-hydra-template

PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation.  ⚡🔥⚡


fast-artistic-videos

Video style transfer using feed-forward networks.


auto-editor

Auto-Editor: Effort free video editing!


jumpcutter

Automatically edits vidx. Explanation here: https://www.youtube.com/watch?v=DQ8orIurGxw


pifuhd

High-Resolution 3D Human Digitization from A Single Image.


kaldi

kaldi-asr/kaldi is the official location of the Kaldi project.


EasySynth

Unreal Engine plugin for easy creation of synthetic image datasets


equivariant-MLP

A library for programmatically generating equivariant layers through constraint solving


TopographicVAE

Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"


ConvNeXt

Code release for ConvNeXt model


spynet

Spatial Pyramid Network for Optical Flow


scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals


delve

PyTorch model training and layer saturation monitor


A toolbox for receptive field analysis and visualizing neural network architectures


cherche

Neural Search


ncnn

ncnn is a high-performance neural network inference framework optimized for the mobile platform


TC-Depth

Self-supervised temporally consistent depth estimation


pims

Python Image Sequence: Load video and sequential images in many formats with a simple, consistent interface.


ru-dolph

RUDOLPH: One Hyper-Tasking Transformer can be creative as DALL-E and GPT-3 and smart as CLIP


tensorboard

TensorFlow's Visualization Toolkit


Interpretability and explainability of data and machine learning models


amundsen

Amundsen is a metadata driven application for improving the productivity of data analysts, data scientists and engineers when interacting with data.


bpycv

Computer vision utils for Blender (generate instance annoatation, depth and 6D pose by one line code)


pypeln

Concurrent data pipelines in Python   >>>


anyio

High level asynchronous concurrency and networking framework that works on top of either trio or asyncio


PytorchWCT

This is the Pytorch implementation of Universal Style Transfer via Feature Transforms.


mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377


Applied-Deep-Learning

Applied Deep Learning Course


fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.


A modular framework for neural networks with Euclidean symmetry


deepchecks

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.


VulkanAutomata

Cellular Automata GPU Renderer using the Vulkan API


Splice

Official Pytorch Implementation for "Splicing ViT Features for Semantic Appearance Transfer" presenting "Splice" (CVPR 2022 Oral)


A series of tutorial notebooks on denoising diffusion probabilistic models in PyTorch


python-blueprint

🐍 Example Python project using best practices 🥇


pyscaffold

🛠 Python project template generator with batteries included


cookiecutter-hypermodern-python

Hypermodern Python Cookiecutter


tyrannosaurus

Generate beautifully modern (2021+) Python projects with seamless, GitHub-based CI/CD, Docker and Conda support, and out-of-the-box integration with >30 best-practices tools and standards.


vision-aided-gan

Ensembling Off-the-shelf Models for GAN Training (CVPR 2022 Oral)


clearml-server

ClearML - Auto-Magical Suite of tools to streamline your ML workflow. Experiment Manager, ML-Ops and Data-Management


prometheus

The Prometheus monitoring system and time series database.


prefect

Prefect is a workflow orchestration tool empowering developers to build, observe, and react to data pipelines


gansformer

Generative Adversarial Transformers


kubric

A data generation pipeline for creating semi-realistic synthetic multi-object videos with rich annotations such as instance segmentation masks, depth maps, and optical flow.


ucsgnet

The official implementation of the "UCSG-Net - Unsupervised Discovering of Constructive Solid Geometry Tree" paper


Ha-NeRF

[CVPR 2022] Ha-NeRF😆: Hallucinated Neural Radiance Fields in the Wild


In-Place Activated BatchNorm for Memory-Optimized Training of DNNs


mvsnerf

[ICCV 2021] Our work presents a novel neural rendering approach that can efficiently reconstruct geometric and neural radiance fields for view synthesis.


cascade-stereo

cascade-stereo


gem

[NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields


awesome-NeRF

A curated list of awesome neural radiance fields papers


awesome-implicit-representations

A curated list of resources on implicit neural representations.


DirectVoxGO

Direct voxel grid optimization for fast radiance field reconstruction.


Plenoxels: Radiance Fields without Neural Networks


nerfplusplus

improves over nerf in 360 capture of unbounded scenes


unisurf

[ICCV'21] UNISURF: Unifying Neural Implicit Surfaces and Radiance Fields for Multi-View Reconstruction


IGR

Implicit Geometric Regularization for Learning Shapes


SCNeRF

[ICCV21] Self-Calibrating Neural Radiance Fields


Thin structure reconstruction


NeuS

Code release for NeuS


Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.


pytorch-minimal-gaussian-process

A minimal implementation of Gaussian process regression in PyTorch


pytorch-ssim

pytorch structural similarity (SSIM) loss


AugLy

A data augmentations library for audio, image, text, and video.


transformer-deploy

Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀


fiftyone

The open-source tool for building high-quality datasets and computer vision models


distributed-ml-patterns

Distributed Machine Learning Patterns from Manning Publications by Yuan Tang https://bit.ly/2RKv8Zo


ai-deadlines

:alarm_clock: AI conference deadline countdowns


kompute

General purpose GPU compute framework built on Vulkan to support 1000s of cross vendor graphics cards (AMD, Qualcomm, NVIDIA & friends). Blazing fast, mobile-enabled, asynchronous and optimized for advanced GPU data processing usecases. Backed by the Linux Foundation.