Beast code in Giters

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Language:PythonApache-2.030355 311 875

gradio

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Language:PythonApache-2.030180 162 4311

OpenVoice

Instant voice cloning by MyShell.

Language:PythonMIT26633 205 196

diffusers

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

Language:PythonApache-2.023307 195 3645

generative-models

Generative Models by Stability AI

Language:PythonMIT22944 248 270

openai-python

The official Python library for the OpenAI API

Language:PythonApache-2.020753 284 685

voice-changer

リアルタイムボイスチェンジャー Realtime Voice Changer

Language:PythonNOASSERTION15060 108 945

CodeFormer

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

Language:PythonNOASSERTION13922 285 321

albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Language:PythonMIT13624 129 937

Swin-Transformer

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

Language:PythonMIT13168 127 302

flash-attention

Fast and memory-efficient exact attention

Language:PythonBSD-3-Clause11523 106 827

triton

Development repository for the Triton language and compiler

Language:C++MIT11235 179 1192

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

Language:Jupyter NotebookMIT10885 97 333

kornia

Geometric Computer Vision Library for Spatial AI

Language:PythonApache-2.09536 129 895

TensorRT

NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source components of TensorRT.

Language:C++Apache-2.09348 150 3499

open_clip

An open source implementation of CLIP.

Language:Jupyter NotebookNOASSERTION8874 77 441

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.

Language:PythonNOASSERTION7858 77 489

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Language:PythonMIT7313 32 273

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.

Language:PythonApache-2.06342 94 542

FasterTransformer

Transformer related optimization, including BERT, GPT

Language:C++Apache-2.05582 65 623

StableSwarmUI

StableSwarmUI, A Modular Stable Diffusion Web-User-Interface, with an emphasis on making powertools easily accessible, high performance, and extensibility.

Language:C#MIT2722 54 304

trt-llm-rag-windows

A developer reference project for creating Retrieval Augmented Generation (RAG) chatbots on Windows using TensorRT-LLM

Language:PythonNOASSERTION2356 46 44

k-diffusion

Karras et al. (2022) diffusion models for PyTorch

Language:PythonMIT2146 41 62

YOLOS

[NeurIPS 2021] You Only Look at One Sequence

Language:Jupyter NotebookMIT810 21 29